gradboard 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

gradboard/__init__.py

@@ -1,3 +1,3 @@
-from . import cycle
+from . import cycles
 from . import optimiser
-from . import schedule
+from . import scheduler

gradboard/{cycle.py → cycles.py}

@@ -188,8 +188,8 @@ class Cycle:
         max_area = 0
         ascent_steps = 0
         descent_steps = 0
-        total_up_gradient = 0
-        total_down_gradient = 0
+        avg_up_gradient = 0
+        avg_down_gradient = 0
         total_gradient = 0
         previous_lr = None
         for s in range(self.total_steps):
@@ -200,11 +200,11 @@ class Cycle:
                 pass
             elif previous_lr > height:
                 descent_steps += 1
-                total_down_gradient += height - previous_lr
+                avg_down_gradient += height - previous_lr
                 total_gradient += height - previous_lr
             elif previous_lr < height:
                 ascent_steps += 1
-                total_up_gradient += height - previous_lr
+                avg_up_gradient += height - previous_lr
                 total_gradient += height - previous_lr
             else:
                 total_gradient += height
@@ -213,12 +213,8 @@ class Cycle:
             "area": total_area / max_area,
             "pc_ascent": round(ascent_steps / self.total_steps, 3),
             "pc_descent": round(descent_steps / self.total_steps, 3),
-            "avg_up_gradient": round(
-                total_up_gradient / ascent_steps if ascent_steps > 0 else 0.0, 3
-            ),
-            "avg_down_gradient": round(
-                total_down_gradient / descent_steps if descent_steps > 0 else 0.0, 3
-            ),
+            "avg_up_gradient": round(avg_up_gradient, 3),
+            "avg_down_gradient": round(avg_down_gradient, 3),
             "avg_gradient": round(-(self.high - self.low) / self.total_steps, 3),
         }
 

gradboard/{schedule.py → scheduler.py}

@@ -10,7 +10,7 @@ from scipy.ndimage import gaussian_filter1d
 
 from torch.amp import GradScaler
 
-from .cycle import Cycle
+from .cycles import Cycle
 
 
 class PASS:

{gradboard-0.1.3.dist-info → gradboard-0.1.5.dist-info}/METADATA

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

gradboard-0.1.5.dist-info/RECORD

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

gradboard-0.1.3.dist-info/RECORD

@@ -1,8 +0,0 @@
-gradboard/__init__.py,sha256=sLjLo-olnO7cbIEphFI9Bc24I4Nv2bw3VfI2m-hjmEs,67
-gradboard/cycle.py,sha256=I0iN0aeckniyK7glVnq3mnVuFE_M30lZCQ09MR2QnOs,9089
-gradboard/optimiser.py,sha256=Iw1piQkQkWdNUW8d9f3iPrvP9qEGVxEKmerK7ATDgOI,6189
-gradboard/schedule.py,sha256=F5L8HT281VuskK2-2dpfP0zBM6XCqxyCXcTc-CDZa1o,6688
-gradboard-0.1.3.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
-gradboard-0.1.3.dist-info/METADATA,sha256=tipi9Kkuz8EY-zNoliGQT_QCdBHkCqi_4vvi2WFqc2s,2173
-gradboard-0.1.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-gradboard-0.1.3.dist-info/RECORD,,