gradboard 0.1.4__tar.gz → 0.1.6__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of gradboard might be problematic. Click here for more details.

@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: gradboard
3
- Version: 0.1.4
3
+ Version: 0.1.6
4
4
  Summary: Easily snowboard down gnarly loss gradients
5
5
  License: MIT
6
6
  Author: Nicholas Bailey
@@ -0,0 +1,3 @@
1
+ from . import cycles
2
+ from . import optimiser
3
+ from . import scheduler
@@ -10,7 +10,7 @@ from scipy.ndimage import gaussian_filter1d
10
10
 
11
11
  from torch.amp import GradScaler
12
12
 
13
- from .cycle import Cycle
13
+ from .cycles import Cycle
14
14
 
15
15
 
16
16
  class PASS:
@@ -28,12 +28,9 @@ class PASS:
28
28
  max_lr: float = None,
29
29
  cool_point: float = None,
30
30
  ):
31
- assert (max_lr is not None) == (cool_point is not None)
32
- assert (
33
- ((max_lr is not None) and (cool_point is not None))
34
- != range_test
35
- is not None
36
- )
31
+ if not range_test:
32
+ assert max_lr is not None
33
+ assert cool_point is not None
37
34
 
38
35
  self.model = model
39
36
  self.optimiser = optimiser
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "gradboard"
3
- version = "0.1.4"
3
+ version = "0.1.6"
4
4
  description = "Easily snowboard down gnarly loss gradients"
5
5
  authors = [
6
6
  {name = "Nicholas Bailey"}
@@ -1,3 +0,0 @@
1
- from . import cycle
2
- from . import optimiser
3
- from . import schedule
File without changes
File without changes