nshtrainer 0.37.0__py3-none-any.whl → 0.38.0__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.
nshtrainer/config.py
CHANGED
|
@@ -65,13 +65,13 @@ from nshtrainer.callbacks.wandb_upload_code import (
|
|
|
65
65
|
WandbUploadCodeConfig as WandbUploadCodeConfig,
|
|
66
66
|
)
|
|
67
67
|
from nshtrainer.callbacks.wandb_watch import WandbWatchConfig as WandbWatchConfig
|
|
68
|
-
from nshtrainer.config import LRSchedulerConfig as LRSchedulerConfig
|
|
69
68
|
from nshtrainer.loggers._base import BaseLoggerConfig as BaseLoggerConfig
|
|
70
69
|
from nshtrainer.loggers.csv import CSVLoggerConfig as CSVLoggerConfig
|
|
71
70
|
from nshtrainer.loggers.tensorboard import (
|
|
72
71
|
TensorboardLoggerConfig as TensorboardLoggerConfig,
|
|
73
72
|
)
|
|
74
73
|
from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
|
|
74
|
+
from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
|
|
75
75
|
from nshtrainer.lr_scheduler._base import LRSchedulerConfigBase as LRSchedulerConfigBase
|
|
76
76
|
from nshtrainer.lr_scheduler.linear_warmup_cosine import (
|
|
77
77
|
DurationConfig as DurationConfig,
|
|
@@ -95,10 +95,10 @@ from nshtrainer.nn.nonlinearity import MishNonlinearityConfig as MishNonlinearit
|
|
|
95
95
|
from nshtrainer.nn.nonlinearity import NonlinearityConfig as NonlinearityConfig
|
|
96
96
|
from nshtrainer.nn.nonlinearity import PReLUConfig as PReLUConfig
|
|
97
97
|
from nshtrainer.nn.nonlinearity import ReLUNonlinearityConfig as ReLUNonlinearityConfig
|
|
98
|
+
from nshtrainer.nn.nonlinearity import SiLUNonlinearityConfig as SiLUNonlinearityConfig
|
|
98
99
|
from nshtrainer.nn.nonlinearity import (
|
|
99
100
|
SigmoidNonlinearityConfig as SigmoidNonlinearityConfig,
|
|
100
101
|
)
|
|
101
|
-
from nshtrainer.nn.nonlinearity import SiLUNonlinearityConfig as SiLUNonlinearityConfig
|
|
102
102
|
from nshtrainer.nn.nonlinearity import (
|
|
103
103
|
SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig,
|
|
104
104
|
)
|
|
@@ -138,12 +138,12 @@ from nshtrainer.trainer._config import ReproducibilityConfig as ReproducibilityC
|
|
|
138
138
|
from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
|
|
139
139
|
from nshtrainer.trainer._config import TrainerConfig as TrainerConfig
|
|
140
140
|
from nshtrainer.util._environment_info import (
|
|
141
|
-
|
|
141
|
+
EnvironmentCUDAConfig as EnvironmentCUDAConfig,
|
|
142
142
|
)
|
|
143
|
-
from nshtrainer.util._environment_info import EnvironmentConfig as EnvironmentConfig
|
|
144
143
|
from nshtrainer.util._environment_info import (
|
|
145
|
-
|
|
144
|
+
EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
|
|
146
145
|
)
|
|
146
|
+
from nshtrainer.util._environment_info import EnvironmentConfig as EnvironmentConfig
|
|
147
147
|
from nshtrainer.util._environment_info import (
|
|
148
148
|
EnvironmentGPUConfig as EnvironmentGPUConfig,
|
|
149
149
|
)
|
|
@@ -151,10 +151,10 @@ from nshtrainer.util._environment_info import (
|
|
|
151
151
|
EnvironmentHardwareConfig as EnvironmentHardwareConfig,
|
|
152
152
|
)
|
|
153
153
|
from nshtrainer.util._environment_info import (
|
|
154
|
-
|
|
154
|
+
EnvironmentLSFInformationConfig as EnvironmentLSFInformationConfig,
|
|
155
155
|
)
|
|
156
156
|
from nshtrainer.util._environment_info import (
|
|
157
|
-
|
|
157
|
+
EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
|
|
158
158
|
)
|
|
159
159
|
from nshtrainer.util._environment_info import (
|
|
160
160
|
EnvironmentPackageConfig as EnvironmentPackageConfig,
|
|
@@ -31,7 +31,7 @@ nshtrainer/callbacks/throughput_monitor.py,sha256=H_ocXErZxUO3dxFk8Tx_VQdpI9E_Zt
|
|
|
31
31
|
nshtrainer/callbacks/timer.py,sha256=quS79oYClDUvQxJkNWmDMe0hwRUkkREgTgqzVrnom50,4607
|
|
32
32
|
nshtrainer/callbacks/wandb_upload_code.py,sha256=OWG4UkL2SfW6oj6AGRXeBJsZmgsqeHLW2Fj8Jm4ga3I,2298
|
|
33
33
|
nshtrainer/callbacks/wandb_watch.py,sha256=Y6SEXfIx3kDDQbI5zpP53BVq0FBLJbLd3RJsiHZk1-Y,2921
|
|
34
|
-
nshtrainer/config.py,sha256=
|
|
34
|
+
nshtrainer/config.py,sha256=HHzDTE-DYDLn7KuX7UHd7JBPVOW_EDiNAUblHPhjpbI,8312
|
|
35
35
|
nshtrainer/data/__init__.py,sha256=7mk1tr7SWUZ7ySbsf0y0ZPszk7u4QznPhQ-7wnpH9ec,149
|
|
36
36
|
nshtrainer/data/balanced_batch_sampler.py,sha256=ybMJF-CguaZ17fLEweZ5suaGOiHOMEm3Bn8rQfGTzGQ,5445
|
|
37
37
|
nshtrainer/data/transform.py,sha256=6SNs3_TpNpfhcwTwvPKyEJ3opM1OT7LmMEYQNHKgRl8,2227
|
|
@@ -99,6 +99,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
|
|
|
99
99
|
nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
|
|
100
100
|
nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
|
|
101
101
|
nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
|
|
102
|
-
nshtrainer-0.
|
|
103
|
-
nshtrainer-0.
|
|
104
|
-
nshtrainer-0.
|
|
102
|
+
nshtrainer-0.38.0.dist-info/METADATA,sha256=0LdRFR5fEfmvO704YWcxFHUunc1A8mVhKTfi_WLjgPI,916
|
|
103
|
+
nshtrainer-0.38.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
|
104
|
+
nshtrainer-0.38.0.dist-info/RECORD,,
|
|
File without changes
|