nshtrainer 0.33.0__py3-none-any.whl → 0.33.2__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/config.py +2 -0
  2. {nshtrainer-0.33.0.dist-info → nshtrainer-0.33.2.dist-info}/METADATA +1 -1
  3. {nshtrainer-0.33.0.dist-info → nshtrainer-0.33.2.dist-info}/RECORD +4 -4
  4. {nshtrainer-0.33.0.dist-info → nshtrainer-0.33.2.dist-info}/WHEEL +0 -0
nshtrainer/config.py CHANGED
@@ -59,6 +59,7 @@ from nshtrainer.loggers.tensorboard import (
59
59
  TensorboardLoggerConfig as TensorboardLoggerConfig,
60
60
  )
61
61
  from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
62
+ from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
62
63
  from nshtrainer.lr_scheduler._base import LRSchedulerConfigBase as LRSchedulerConfigBase
63
64
  from nshtrainer.lr_scheduler.linear_warmup_cosine import (
64
65
  DurationConfig as DurationConfig,
@@ -103,6 +104,7 @@ from nshtrainer.nn.nonlinearity import (
103
104
  )
104
105
  from nshtrainer.nn.nonlinearity import TanhNonlinearityConfig as TanhNonlinearityConfig
105
106
  from nshtrainer.optimizer import AdamWConfig as AdamWConfig
107
+ from nshtrainer.optimizer import OptimizerConfig as OptimizerConfig
106
108
  from nshtrainer.optimizer import OptimizerConfigBase as OptimizerConfigBase
107
109
  from nshtrainer.profiler._base import BaseProfilerConfig as BaseProfilerConfig
108
110
  from nshtrainer.profiler.advanced import (
{nshtrainer-0.33.0.dist-info → nshtrainer-0.33.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.33.0
3
+ Version: 0.33.2
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.33.0.dist-info → nshtrainer-0.33.2.dist-info}/RECORD RENAMED
@@ -30,7 +30,7 @@ nshtrainer/callbacks/shared_parameters.py,sha256=fqlDweFDXPV_bfcAWpRgaJIad9i5Aeh
30
30
  nshtrainer/callbacks/throughput_monitor.py,sha256=H_ocXErZxUO3dxFk8Tx_VQdpI9E_Ztvqof5WtFevLyQ,1838
31
31
  nshtrainer/callbacks/timer.py,sha256=quS79oYClDUvQxJkNWmDMe0hwRUkkREgTgqzVrnom50,4607
32
32
  nshtrainer/callbacks/wandb_watch.py,sha256=Y6SEXfIx3kDDQbI5zpP53BVq0FBLJbLd3RJsiHZk1-Y,2921
33
- nshtrainer/config.py,sha256=tS3Qw328weUEe3Of548NY1hiecfLbA60HsyKLkc24Hc,6740
33
+ nshtrainer/config.py,sha256=EqvSp06RmSkCvo13-5bkecoCcE1nVViwvFIivTZOXoI,6883
34
34
  nshtrainer/data/__init__.py,sha256=7mk1tr7SWUZ7ySbsf0y0ZPszk7u4QznPhQ-7wnpH9ec,149
35
35
  nshtrainer/data/balanced_batch_sampler.py,sha256=dGBTDDtlBU6c-ZlVQOCnTW7SjTB5hczWsOWEdUWjvkA,4385
36
36
  nshtrainer/data/transform.py,sha256=6SNs3_TpNpfhcwTwvPKyEJ3opM1OT7LmMEYQNHKgRl8,2227
@@ -95,6 +95,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
95
95
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
96
96
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
97
97
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
98
- nshtrainer-0.33.0.dist-info/METADATA,sha256=vQQtxiOWksdMxSZai1mmDNOzVNF8N4sAqsW3Ji46HDg,916
99
- nshtrainer-0.33.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
100
- nshtrainer-0.33.0.dist-info/RECORD,,
98
+ nshtrainer-0.33.2.dist-info/METADATA,sha256=zDC_xehJGE3RlCACScFpu64qL1TKd_D8VyhjmRxNDkw,916
99
+ nshtrainer-0.33.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
100
+ nshtrainer-0.33.2.dist-info/RECORD,,