gradboard 0.1.10__tar.gz → 0.1.11__tar.gz

{gradboard-0.1.10 → gradboard-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: gradboard
-Version: 0.1.10
+Version: 0.1.11
 Summary: Easily snowboard down gnarly loss gradients
 License: MIT
 Author: Nicholas Bailey

{gradboard-0.1.10 → gradboard-0.1.11}/gradboard/cycles.py

@@ -221,7 +221,12 @@ class Cycle:
 
 
 class CycleProduct(Cycle):
-    def __init__(self, cycles: List[Cycle], reflect=False):
+    def __init__(self, cycles: List[Cycle], reflect=False, normalise: bool = False):
+        """
+        Args:
+            normalise: if true, the square root of the product is returned (i.e.
+                the geometric mean of the two cycles that were multiplied together)
+        """
         main_training_examples = cycles[0].training_examples
         main_batch_size = cycles[0].batch_size
 
@@ -230,11 +235,14 @@ class CycleProduct(Cycle):
 
         self.cycles = cycles
         self.reflect = reflect
+        self.normalise = normalise
 
         def generating_function(step: int, total_steps: int) -> float:
             output = self.cycles[0](step)
             for c in self.cycles[1:]:
                 output *= c(step % c.total_steps)
+            if self.normalise:
+                output = math.sqrt(output)
             return output
 
         super().__init__(

{gradboard-0.1.10 → gradboard-0.1.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "gradboard"
-version = "0.1.10"
+version = "0.1.11"
 description = "Easily snowboard down gnarly loss gradients"
 authors = [
     {name = "Nicholas Bailey"}