gradboard 0.1.8__tar.gz → 0.1.9__tar.gz

{gradboard-0.1.8 → gradboard-0.1.9}/PKG-INFO

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

{gradboard-0.1.8 → gradboard-0.1.9}/gradboard/cycles.py

@@ -178,6 +178,7 @@ class Cycle:
 
         return list(zip(widths, heights, strict=True))
 
+    @property
     def stats(self) -> float:
         """
         Returns the area (as a percentage of the area of a curve where the learning
@@ -271,3 +272,18 @@ class CycleSequence:
 
     def __len__(self):
         return self.total_steps
+
+    @property
+    def stats(self) -> float:
+        """
+        Returns the area (as a percentage of the area of a curve where the learning
+            rate is constant max_lr), percentage ascent steps and percentage descent
+            steps of a learning rate schedule.
+        """
+        cycle_ratios = [c.total_steps for c in self.cycles]
+        cycle_stats = {k: v * cycle_ratios[0] for k, v in self.cycles[0].stats.items()}
+        for i, cycle in enumerate(self.cycles[1:]):
+            for k, v in cycle.stats.items():
+                cycle_stats[k] += cycle_ratios[i] * v
+
+        return {k: v / self.total_steps for k, v in cycle_stats.items()}

{gradboard-0.1.8 → gradboard-0.1.9}/pyproject.toml

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