gradboard 0.1.8__py3-none-any.whl → 0.1.10__py3-none-any.whl

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,20 @@ 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):
+            if i == 0:
+                continue  # We already did the first one, above
+            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.dist-info → gradboard-0.1.10.dist-info}/METADATA

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

gradboard-0.1.10.dist-info/RECORD

@@ -0,0 +1,8 @@
+gradboard/__init__.py,sha256=57AkHusYwLCsusiVnajH5pMFKioRCj-3IjF9qpdOzE0,69
+gradboard/cycles.py,sha256=dy3qr4kUPPmB0g3L-E1CMYZSuw6son6fIvyu9uUJglQ,9682
+gradboard/optimiser.py,sha256=Iw1piQkQkWdNUW8d9f3iPrvP9qEGVxEKmerK7ATDgOI,6189
+gradboard/scheduler.py,sha256=W146MgdTYz113R3M0qhVpk8kr6L1ioPS3pCFc3MhKYg,6507
+gradboard-0.1.10.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
+gradboard-0.1.10.dist-info/METADATA,sha256=4ygT3hkBfJWE0jP27o0CxcCwW86x69V2n3kKYMzkupU,2174
+gradboard-0.1.10.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+gradboard-0.1.10.dist-info/RECORD,,

gradboard-0.1.8.dist-info/RECORD

@@ -1,8 +0,0 @@
-gradboard/__init__.py,sha256=57AkHusYwLCsusiVnajH5pMFKioRCj-3IjF9qpdOzE0,69
-gradboard/cycles.py,sha256=8mdqRDCVI6kUEqCcLOo99pzwMngsRNZI-JwrYucpadQ,8927
-gradboard/optimiser.py,sha256=Iw1piQkQkWdNUW8d9f3iPrvP9qEGVxEKmerK7ATDgOI,6189
-gradboard/scheduler.py,sha256=W146MgdTYz113R3M0qhVpk8kr6L1ioPS3pCFc3MhKYg,6507
-gradboard-0.1.8.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
-gradboard-0.1.8.dist-info/METADATA,sha256=F286NyzEeIX1TKEhh8otKWko_jtCeHBsIwJKE5pi-vw,2173
-gradboard-0.1.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-gradboard-0.1.8.dist-info/RECORD,,