nshtrainer 0.30.0__py3-none-any.whl → 0.30.1__py3-none-any.whl

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.
Files changed (4) hide show
  1. nshtrainer/lr_scheduler/linear_warmup_cosine.py +2 -7
  2. {nshtrainer-0.30.0.dist-info → nshtrainer-0.30.1.dist-info}/METADATA +1 -1
  3. {nshtrainer-0.30.0.dist-info → nshtrainer-0.30.1.dist-info}/RECORD +4 -4
  4. {nshtrainer-0.30.0.dist-info → nshtrainer-0.30.1.dist-info}/WHEEL +0 -0
nshtrainer/lr_scheduler/linear_warmup_cosine.py CHANGED
@@ -121,13 +121,8 @@ class LinearWarmupCosineDecayLRSchedulerConfig(LRSchedulerConfigBase):
121
121
  @override
122
122
  def create_scheduler_impl(self, optimizer, lightning_module, lr):
123
123
  num_steps_per_epoch = self.compute_num_steps_per_epoch(lightning_module)
124
- warmup_steps = (
125
- self.warmup_duration.to_steps(num_steps_per_epoch).value
126
- * num_steps_per_epoch
127
- )
128
- max_steps = (
129
- self.max_duration.to_steps(num_steps_per_epoch).value * num_steps_per_epoch
130
- )
124
+ warmup_steps = self.warmup_duration.to_steps(num_steps_per_epoch).value
125
+ max_steps = self.max_duration.to_steps(num_steps_per_epoch).value
131
126
  warmup_start_lr = self.warmup_start_lr_factor * lr
132
127
  min_lr = self.min_lr_factor * lr
133
128
 
{nshtrainer-0.30.0.dist-info → nshtrainer-0.30.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.30.0
3
+ Version: 0.30.1
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.30.0.dist-info → nshtrainer-0.30.1.dist-info}/RECORD RENAMED
@@ -54,7 +54,7 @@ nshtrainer/loggers/tensorboard.py,sha256=wL2amRSdP68zbslZvBeM0ZQBnjF3hIKsz-_lBbd
54
54
  nshtrainer/loggers/wandb.py,sha256=FPwbf618AYmuPzHdhd1ZFhJ8qDjwTUiSe7cm7g3KCyM,5112
55
55
  nshtrainer/lr_scheduler/__init__.py,sha256=uEvgaFAs-4s_bAEMaildy0GT6OvgpgOEKTuzqutESHE,736
56
56
  nshtrainer/lr_scheduler/_base.py,sha256=7xOIuxQ86YHbFWG5a3gX46emQj1WN_LaY4-i0Q1TDBg,3659
57
- nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=pmX5n7mmhSqPTz4Nu9g_JTsE9gzCkuU4V3GuAHUsDoA,5451
57
+ nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=Fyontbfu4k2932xZenE63QL4CrVGWANXdTeq63dUko0,5347
58
58
  nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=h76oTHYpMxauV_l6lviya5DW-WKArwxxf7ZQizhmbCw,2782
59
59
  nshtrainer/metrics/__init__.py,sha256=ObLIELGguIEcUpRsUkqh1ltrvZii6vglTpJGrPvoy00,50
60
60
  nshtrainer/metrics/_config.py,sha256=jgRBfDAQLFTW7AiUY7CRtdfts6CR6keeuqm0FFMWCzQ,1288
@@ -89,6 +89,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
89
89
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
90
90
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
91
91
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
92
- nshtrainer-0.30.0.dist-info/METADATA,sha256=lDudS-lD7exw8lNe_3vT13ysnk491QCkObXGLQtjhMk,916
93
- nshtrainer-0.30.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
94
- nshtrainer-0.30.0.dist-info/RECORD,,
92
+ nshtrainer-0.30.1.dist-info/METADATA,sha256=LV0wQlmotpfC3qO76dFVCbS26bEl-9YMiTetEeqVQsU,916
93
+ nshtrainer-0.30.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
94
+ nshtrainer-0.30.1.dist-info/RECORD,,