gradboard 1.0.2__py3-none-any.whl → 2.1.0__py3-none-any.whl

gradboard/scheduler.py

@@ -159,10 +159,23 @@ class PASS:
         """
         range_test_results = self._smoothed_range_test(self.range_test_results)
         minimum = min(range_test_results, key=lambda x: x[1])
+        max_left_of_min = (0, 0)
         for r in range_test_results:
-            if (r[0] < minimum[0]) and (r[1] > minimum[1]):
-                self.max_lr = r[0] / 2
-                self.cool_point = self.max_lr * 0.1
+            if r[0] > minimum[0]:
+                break
+            elif r[1] > max_left_of_min[1]:
+                max_left_of_min = r
+            else:
+                continue
+        difference = max_left_of_min[1] - minimum[1]
+        for r in range_test_results:
+            if r[1] > minimum[1] + 0.66 * difference:
+                self.cool_point = r[0]
+            elif r[1] > minimum[1] + 0.33 * difference:
+                self.max_lr = r[0]
+            else:
+                continue
+        self.cool_point = min(self.cool_point, self.max_lr / 4)
         print("High LR", self.max_lr)
         print("Cool point", self.cool_point)
 

{gradboard-1.0.2.dist-info → gradboard-2.1.0.dist-info}/METADATA

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

gradboard-2.1.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+gradboard/__init__.py,sha256=57AkHusYwLCsusiVnajH5pMFKioRCj-3IjF9qpdOzE0,69
+gradboard/cycles.py,sha256=iGEW3Rlp-JNyQZLfpaDXxUCAcEV01ANjLF-Fnhug-qA,10120
+gradboard/optimiser.py,sha256=h5nntenAhD8E9-Ayt17r7AK42S7O-XdhsrxedDd_4co,6201
+gradboard/scheduler.py,sha256=hwvemOp6I3hbf3PWE9aag3c2myQ2Q93V5Sg6K0qqT7s,6721
+gradboard-2.1.0.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
+gradboard-2.1.0.dist-info/METADATA,sha256=aT2hPoM5Z9BzQyMjjAjmoZbuqP2njmoHAHK--XM__wA,2173
+gradboard-2.1.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+gradboard-2.1.0.dist-info/RECORD,,

gradboard-1.0.2.dist-info/RECORD

@@ -1,8 +0,0 @@
-gradboard/__init__.py,sha256=57AkHusYwLCsusiVnajH5pMFKioRCj-3IjF9qpdOzE0,69
-gradboard/cycles.py,sha256=iGEW3Rlp-JNyQZLfpaDXxUCAcEV01ANjLF-Fnhug-qA,10120
-gradboard/optimiser.py,sha256=h5nntenAhD8E9-Ayt17r7AK42S7O-XdhsrxedDd_4co,6201
-gradboard/scheduler.py,sha256=icIDEcem0jwB-NORUsaZvFNjOOn52R5g8OFBD4CYzsE,6279
-gradboard-1.0.2.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
-gradboard-1.0.2.dist-info/METADATA,sha256=ZVjXU2FrPsPtuX1-OPXlZY97F4E0aaRGpOk3Uhps8BQ,2173
-gradboard-1.0.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-gradboard-1.0.2.dist-info/RECORD,,