nshtrainer 0.41.0__py3-none-any.whl → 0.42.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/__init__.py +1 -2
- nshtrainer/config/__init__.py +406 -95
- nshtrainer/config/_checkpoint/loader/__init__.py +55 -13
- nshtrainer/config/_checkpoint/metadata/__init__.py +22 -8
- nshtrainer/config/_directory/__init__.py +21 -9
- nshtrainer/config/_hf_hub/__init__.py +25 -9
- nshtrainer/config/callbacks/__init__.py +114 -29
- nshtrainer/config/callbacks/actsave/__init__.py +20 -8
- nshtrainer/config/callbacks/base/__init__.py +17 -7
- nshtrainer/config/callbacks/checkpoint/__init__.py +62 -13
- nshtrainer/config/callbacks/checkpoint/_base/__init__.py +33 -9
- nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py +40 -10
- nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py +33 -9
- nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py +26 -8
- nshtrainer/config/callbacks/debug_flag/__init__.py +24 -8
- nshtrainer/config/callbacks/directory_setup/__init__.py +26 -8
- nshtrainer/config/callbacks/early_stopping/__init__.py +31 -9
- nshtrainer/config/callbacks/ema/__init__.py +20 -8
- nshtrainer/config/callbacks/finite_checks/__init__.py +26 -8
- nshtrainer/config/callbacks/gradient_skipping/__init__.py +26 -8
- nshtrainer/config/callbacks/norm_logging/__init__.py +24 -8
- nshtrainer/config/callbacks/print_table/__init__.py +26 -8
- nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py +26 -8
- nshtrainer/config/callbacks/shared_parameters/__init__.py +26 -8
- nshtrainer/config/callbacks/throughput_monitor/__init__.py +26 -8
- nshtrainer/config/callbacks/timer/__init__.py +22 -8
- nshtrainer/config/callbacks/wandb_upload_code/__init__.py +26 -8
- nshtrainer/config/callbacks/wandb_watch/__init__.py +24 -8
- nshtrainer/config/loggers/__init__.py +41 -14
- nshtrainer/config/loggers/_base/__init__.py +15 -7
- nshtrainer/config/loggers/csv/__init__.py +18 -8
- nshtrainer/config/loggers/tensorboard/__init__.py +24 -8
- nshtrainer/config/loggers/wandb/__init__.py +31 -11
- nshtrainer/config/lr_scheduler/__init__.py +49 -12
- nshtrainer/config/lr_scheduler/_base/__init__.py +19 -7
- nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py +33 -9
- nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py +33 -9
- nshtrainer/config/metrics/__init__.py +16 -7
- nshtrainer/config/metrics/_config/__init__.py +15 -7
- nshtrainer/config/model/__init__.py +31 -12
- nshtrainer/config/model/base/__init__.py +18 -8
- nshtrainer/config/model/config/__init__.py +30 -12
- nshtrainer/config/model/mixins/logger/__init__.py +15 -7
- nshtrainer/config/nn/__init__.py +68 -23
- nshtrainer/config/nn/mlp/__init__.py +21 -9
- nshtrainer/config/nn/nonlinearity/__init__.py +118 -22
- nshtrainer/config/optimizer/__init__.py +21 -9
- nshtrainer/config/profiler/__init__.py +28 -11
- nshtrainer/config/profiler/_base/__init__.py +17 -7
- nshtrainer/config/profiler/advanced/__init__.py +24 -8
- nshtrainer/config/profiler/pytorch/__init__.py +24 -8
- nshtrainer/config/profiler/simple/__init__.py +22 -8
- nshtrainer/config/runner/__init__.py +15 -7
- nshtrainer/config/trainer/_config/__init__.py +144 -30
- nshtrainer/config/trainer/checkpoint_connector/__init__.py +19 -7
- nshtrainer/config/util/_environment_info/__init__.py +87 -17
- nshtrainer/config/util/config/__init__.py +25 -10
- nshtrainer/config/util/config/dtype/__init__.py +15 -7
- nshtrainer/config/util/config/duration/__init__.py +27 -9
- {nshtrainer-0.41.0.dist-info → nshtrainer-0.42.0.dist-info}/METADATA +1 -1
- {nshtrainer-0.41.0.dist-info → nshtrainer-0.42.0.dist-info}/RECORD +62 -62
- {nshtrainer-0.41.0.dist-info → nshtrainer-0.42.0.dist-info}/WHEEL +0 -0
|
@@ -1,22 +1,92 @@
|
|
|
1
|
-
# fmt: off
|
|
2
|
-
# ruff: noqa
|
|
3
|
-
# type: ignore
|
|
4
|
-
|
|
5
1
|
__codegen__ = True
|
|
6
2
|
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
from nshtrainer.util._environment_info import
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
from nshtrainer.util._environment_info import
|
|
16
|
-
from nshtrainer.util._environment_info import
|
|
17
|
-
|
|
18
|
-
|
|
3
|
+
from typing import TYPE_CHECKING
|
|
4
|
+
|
|
5
|
+
# Config/alias imports
|
|
6
|
+
|
|
7
|
+
if TYPE_CHECKING:
|
|
8
|
+
from nshtrainer.util._environment_info import (
|
|
9
|
+
EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
|
|
10
|
+
)
|
|
11
|
+
from nshtrainer.util._environment_info import EnvironmentConfig as EnvironmentConfig
|
|
12
|
+
from nshtrainer.util._environment_info import (
|
|
13
|
+
EnvironmentCUDAConfig as EnvironmentCUDAConfig,
|
|
14
|
+
)
|
|
15
|
+
from nshtrainer.util._environment_info import (
|
|
16
|
+
EnvironmentGPUConfig as EnvironmentGPUConfig,
|
|
17
|
+
)
|
|
18
|
+
from nshtrainer.util._environment_info import (
|
|
19
|
+
EnvironmentHardwareConfig as EnvironmentHardwareConfig,
|
|
20
|
+
)
|
|
21
|
+
from nshtrainer.util._environment_info import (
|
|
22
|
+
EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
|
|
23
|
+
)
|
|
24
|
+
from nshtrainer.util._environment_info import (
|
|
25
|
+
EnvironmentLSFInformationConfig as EnvironmentLSFInformationConfig,
|
|
26
|
+
)
|
|
27
|
+
from nshtrainer.util._environment_info import (
|
|
28
|
+
EnvironmentPackageConfig as EnvironmentPackageConfig,
|
|
29
|
+
)
|
|
30
|
+
from nshtrainer.util._environment_info import (
|
|
31
|
+
EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig,
|
|
32
|
+
)
|
|
33
|
+
from nshtrainer.util._environment_info import (
|
|
34
|
+
EnvironmentSnapshotConfig as EnvironmentSnapshotConfig,
|
|
35
|
+
)
|
|
36
|
+
from nshtrainer.util._environment_info import (
|
|
37
|
+
GitRepositoryConfig as GitRepositoryConfig,
|
|
38
|
+
)
|
|
39
|
+
else:
|
|
40
|
+
|
|
41
|
+
def __getattr__(name):
|
|
42
|
+
import importlib
|
|
19
43
|
|
|
20
|
-
|
|
44
|
+
if name in globals():
|
|
45
|
+
return globals()[name]
|
|
46
|
+
if name == "EnvironmentPackageConfig":
|
|
47
|
+
return importlib.import_module(
|
|
48
|
+
"nshtrainer.util._environment_info"
|
|
49
|
+
).EnvironmentPackageConfig
|
|
50
|
+
if name == "EnvironmentSnapshotConfig":
|
|
51
|
+
return importlib.import_module(
|
|
52
|
+
"nshtrainer.util._environment_info"
|
|
53
|
+
).EnvironmentSnapshotConfig
|
|
54
|
+
if name == "EnvironmentLSFInformationConfig":
|
|
55
|
+
return importlib.import_module(
|
|
56
|
+
"nshtrainer.util._environment_info"
|
|
57
|
+
).EnvironmentLSFInformationConfig
|
|
58
|
+
if name == "EnvironmentLinuxEnvironmentConfig":
|
|
59
|
+
return importlib.import_module(
|
|
60
|
+
"nshtrainer.util._environment_info"
|
|
61
|
+
).EnvironmentLinuxEnvironmentConfig
|
|
62
|
+
if name == "EnvironmentSLURMInformationConfig":
|
|
63
|
+
return importlib.import_module(
|
|
64
|
+
"nshtrainer.util._environment_info"
|
|
65
|
+
).EnvironmentSLURMInformationConfig
|
|
66
|
+
if name == "EnvironmentConfig":
|
|
67
|
+
return importlib.import_module(
|
|
68
|
+
"nshtrainer.util._environment_info"
|
|
69
|
+
).EnvironmentConfig
|
|
70
|
+
if name == "EnvironmentClassInformationConfig":
|
|
71
|
+
return importlib.import_module(
|
|
72
|
+
"nshtrainer.util._environment_info"
|
|
73
|
+
).EnvironmentClassInformationConfig
|
|
74
|
+
if name == "GitRepositoryConfig":
|
|
75
|
+
return importlib.import_module(
|
|
76
|
+
"nshtrainer.util._environment_info"
|
|
77
|
+
).GitRepositoryConfig
|
|
78
|
+
if name == "EnvironmentCUDAConfig":
|
|
79
|
+
return importlib.import_module(
|
|
80
|
+
"nshtrainer.util._environment_info"
|
|
81
|
+
).EnvironmentCUDAConfig
|
|
82
|
+
if name == "EnvironmentGPUConfig":
|
|
83
|
+
return importlib.import_module(
|
|
84
|
+
"nshtrainer.util._environment_info"
|
|
85
|
+
).EnvironmentGPUConfig
|
|
86
|
+
if name == "EnvironmentHardwareConfig":
|
|
87
|
+
return importlib.import_module(
|
|
88
|
+
"nshtrainer.util._environment_info"
|
|
89
|
+
).EnvironmentHardwareConfig
|
|
90
|
+
raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
|
|
21
91
|
|
|
22
92
|
# Submodule exports
|
|
@@ -1,16 +1,31 @@
|
|
|
1
|
-
# fmt: off
|
|
2
|
-
# ruff: noqa
|
|
3
|
-
# type: ignore
|
|
4
|
-
|
|
5
1
|
__codegen__ = True
|
|
6
2
|
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
3
|
+
from typing import TYPE_CHECKING
|
|
4
|
+
|
|
5
|
+
# Config/alias imports
|
|
6
|
+
|
|
7
|
+
if TYPE_CHECKING:
|
|
8
|
+
from nshtrainer.util.config import DTypeConfig as DTypeConfig
|
|
9
|
+
from nshtrainer.util.config import DurationConfig as DurationConfig
|
|
10
|
+
from nshtrainer.util.config import EpochsConfig as EpochsConfig
|
|
11
|
+
from nshtrainer.util.config import StepsConfig as StepsConfig
|
|
12
|
+
else:
|
|
13
|
+
|
|
14
|
+
def __getattr__(name):
|
|
15
|
+
import importlib
|
|
16
|
+
|
|
17
|
+
if name in globals():
|
|
18
|
+
return globals()[name]
|
|
19
|
+
if name == "EpochsConfig":
|
|
20
|
+
return importlib.import_module("nshtrainer.util.config").EpochsConfig
|
|
21
|
+
if name == "StepsConfig":
|
|
22
|
+
return importlib.import_module("nshtrainer.util.config").StepsConfig
|
|
23
|
+
if name == "DTypeConfig":
|
|
24
|
+
return importlib.import_module("nshtrainer.util.config").DTypeConfig
|
|
25
|
+
if name == "DurationConfig":
|
|
26
|
+
return importlib.import_module("nshtrainer.util.config").DurationConfig
|
|
27
|
+
raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
|
|
11
28
|
|
|
12
|
-
# Type aliases
|
|
13
|
-
from nshtrainer.util.config import DurationConfig as DurationConfig
|
|
14
29
|
|
|
15
30
|
# Submodule exports
|
|
16
31
|
from . import dtype as dtype
|
|
@@ -1,12 +1,20 @@
|
|
|
1
|
-
# fmt: off
|
|
2
|
-
# ruff: noqa
|
|
3
|
-
# type: ignore
|
|
4
|
-
|
|
5
1
|
__codegen__ = True
|
|
6
2
|
|
|
7
|
-
|
|
8
|
-
|
|
3
|
+
from typing import TYPE_CHECKING
|
|
4
|
+
|
|
5
|
+
# Config/alias imports
|
|
6
|
+
|
|
7
|
+
if TYPE_CHECKING:
|
|
8
|
+
from nshtrainer.util.config.dtype import DTypeConfig as DTypeConfig
|
|
9
|
+
else:
|
|
10
|
+
|
|
11
|
+
def __getattr__(name):
|
|
12
|
+
import importlib
|
|
9
13
|
|
|
10
|
-
|
|
14
|
+
if name in globals():
|
|
15
|
+
return globals()[name]
|
|
16
|
+
if name == "DTypeConfig":
|
|
17
|
+
return importlib.import_module("nshtrainer.util.config.dtype").DTypeConfig
|
|
18
|
+
raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
|
|
11
19
|
|
|
12
20
|
# Submodule exports
|
|
@@ -1,14 +1,32 @@
|
|
|
1
|
-
# fmt: off
|
|
2
|
-
# ruff: noqa
|
|
3
|
-
# type: ignore
|
|
4
|
-
|
|
5
1
|
__codegen__ = True
|
|
6
2
|
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
3
|
+
from typing import TYPE_CHECKING
|
|
4
|
+
|
|
5
|
+
# Config/alias imports
|
|
6
|
+
|
|
7
|
+
if TYPE_CHECKING:
|
|
8
|
+
from nshtrainer.util.config.duration import DurationConfig as DurationConfig
|
|
9
|
+
from nshtrainer.util.config.duration import EpochsConfig as EpochsConfig
|
|
10
|
+
from nshtrainer.util.config.duration import StepsConfig as StepsConfig
|
|
11
|
+
else:
|
|
12
|
+
|
|
13
|
+
def __getattr__(name):
|
|
14
|
+
import importlib
|
|
10
15
|
|
|
11
|
-
|
|
12
|
-
|
|
16
|
+
if name in globals():
|
|
17
|
+
return globals()[name]
|
|
18
|
+
if name == "EpochsConfig":
|
|
19
|
+
return importlib.import_module(
|
|
20
|
+
"nshtrainer.util.config.duration"
|
|
21
|
+
).EpochsConfig
|
|
22
|
+
if name == "StepsConfig":
|
|
23
|
+
return importlib.import_module(
|
|
24
|
+
"nshtrainer.util.config.duration"
|
|
25
|
+
).StepsConfig
|
|
26
|
+
if name == "DurationConfig":
|
|
27
|
+
return importlib.import_module(
|
|
28
|
+
"nshtrainer.util.config.duration"
|
|
29
|
+
).DurationConfig
|
|
30
|
+
raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
|
|
13
31
|
|
|
14
32
|
# Submodule exports
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
nshtrainer/__init__.py,sha256=
|
|
1
|
+
nshtrainer/__init__.py,sha256=8hx3uqzroMYOQM7S4U3Eznw7MHF3YDqUkO1MDQeC4cM,642
|
|
2
2
|
nshtrainer/_callback.py,sha256=A1zLsTy4b_wOYnInLLXGSRdHzT2yNa6mPEql-ozm0u0,1013
|
|
3
3
|
nshtrainer/_checkpoint/loader.py,sha256=5vjg-OFChXJjgiOVv8vnV8nwTscfdDtEdxQRz6uPfDE,14158
|
|
4
4
|
nshtrainer/_checkpoint/metadata.py,sha256=5D4PgKodzhLsmQvuF3xxkH49epKaegxi4wh_ImDTtns,4737
|
|
@@ -31,64 +31,64 @@ 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/__init__.py,sha256=
|
|
35
|
-
nshtrainer/config/_checkpoint/loader/__init__.py,sha256=
|
|
36
|
-
nshtrainer/config/_checkpoint/metadata/__init__.py,sha256=
|
|
37
|
-
nshtrainer/config/_directory/__init__.py,sha256=
|
|
38
|
-
nshtrainer/config/_hf_hub/__init__.py,sha256=
|
|
39
|
-
nshtrainer/config/callbacks/__init__.py,sha256=
|
|
40
|
-
nshtrainer/config/callbacks/actsave/__init__.py,sha256=
|
|
41
|
-
nshtrainer/config/callbacks/base/__init__.py,sha256=
|
|
42
|
-
nshtrainer/config/callbacks/checkpoint/__init__.py,sha256=
|
|
43
|
-
nshtrainer/config/callbacks/checkpoint/_base/__init__.py,sha256=
|
|
44
|
-
nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=
|
|
45
|
-
nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=
|
|
46
|
-
nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=
|
|
47
|
-
nshtrainer/config/callbacks/debug_flag/__init__.py,sha256=
|
|
48
|
-
nshtrainer/config/callbacks/directory_setup/__init__.py,sha256=
|
|
49
|
-
nshtrainer/config/callbacks/early_stopping/__init__.py,sha256=
|
|
50
|
-
nshtrainer/config/callbacks/ema/__init__.py,sha256=
|
|
51
|
-
nshtrainer/config/callbacks/finite_checks/__init__.py,sha256=
|
|
52
|
-
nshtrainer/config/callbacks/gradient_skipping/__init__.py,sha256=
|
|
53
|
-
nshtrainer/config/callbacks/norm_logging/__init__.py,sha256=
|
|
54
|
-
nshtrainer/config/callbacks/print_table/__init__.py,sha256=
|
|
55
|
-
nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py,sha256
|
|
56
|
-
nshtrainer/config/callbacks/shared_parameters/__init__.py,sha256=
|
|
57
|
-
nshtrainer/config/callbacks/throughput_monitor/__init__.py,sha256=
|
|
58
|
-
nshtrainer/config/callbacks/timer/__init__.py,sha256=
|
|
59
|
-
nshtrainer/config/callbacks/wandb_upload_code/__init__.py,sha256=
|
|
60
|
-
nshtrainer/config/callbacks/wandb_watch/__init__.py,sha256=
|
|
61
|
-
nshtrainer/config/loggers/__init__.py,sha256=
|
|
62
|
-
nshtrainer/config/loggers/_base/__init__.py,sha256=
|
|
63
|
-
nshtrainer/config/loggers/csv/__init__.py,sha256=
|
|
64
|
-
nshtrainer/config/loggers/tensorboard/__init__.py,sha256=
|
|
65
|
-
nshtrainer/config/loggers/wandb/__init__.py,sha256=
|
|
66
|
-
nshtrainer/config/lr_scheduler/__init__.py,sha256=
|
|
67
|
-
nshtrainer/config/lr_scheduler/_base/__init__.py,sha256=
|
|
68
|
-
nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py,sha256=
|
|
69
|
-
nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py,sha256=
|
|
70
|
-
nshtrainer/config/metrics/__init__.py,sha256=
|
|
71
|
-
nshtrainer/config/metrics/_config/__init__.py,sha256=
|
|
72
|
-
nshtrainer/config/model/__init__.py,sha256=
|
|
73
|
-
nshtrainer/config/model/base/__init__.py,sha256=
|
|
74
|
-
nshtrainer/config/model/config/__init__.py,sha256=
|
|
75
|
-
nshtrainer/config/model/mixins/logger/__init__.py,sha256=
|
|
76
|
-
nshtrainer/config/nn/__init__.py,sha256=
|
|
77
|
-
nshtrainer/config/nn/mlp/__init__.py,sha256=
|
|
78
|
-
nshtrainer/config/nn/nonlinearity/__init__.py,sha256=
|
|
79
|
-
nshtrainer/config/optimizer/__init__.py,sha256=
|
|
80
|
-
nshtrainer/config/profiler/__init__.py,sha256=
|
|
81
|
-
nshtrainer/config/profiler/_base/__init__.py,sha256=
|
|
82
|
-
nshtrainer/config/profiler/advanced/__init__.py,sha256=
|
|
83
|
-
nshtrainer/config/profiler/pytorch/__init__.py,sha256=
|
|
84
|
-
nshtrainer/config/profiler/simple/__init__.py,sha256=
|
|
85
|
-
nshtrainer/config/runner/__init__.py,sha256=
|
|
86
|
-
nshtrainer/config/trainer/_config/__init__.py,sha256=
|
|
87
|
-
nshtrainer/config/trainer/checkpoint_connector/__init__.py,sha256=
|
|
88
|
-
nshtrainer/config/util/_environment_info/__init__.py,sha256=
|
|
89
|
-
nshtrainer/config/util/config/__init__.py,sha256=
|
|
90
|
-
nshtrainer/config/util/config/dtype/__init__.py,sha256=
|
|
91
|
-
nshtrainer/config/util/config/duration/__init__.py,sha256=
|
|
34
|
+
nshtrainer/config/__init__.py,sha256=fFp5PCXKVVJqiv4gONdo4fpDNReX8uSQTWliUQmBa60,21848
|
|
35
|
+
nshtrainer/config/_checkpoint/loader/__init__.py,sha256=hCjXcva0NMMezDumPYoBxKdXkNv8IhALgGforQjUbWw,2363
|
|
36
|
+
nshtrainer/config/_checkpoint/metadata/__init__.py,sha256=oCpMBf7PduAaK6bxcNqdzVNdCWR2kvbOEvbgIKo166M,831
|
|
37
|
+
nshtrainer/config/_directory/__init__.py,sha256=I5CO98ktOWzCGmtcYlWYt3Fa0BwK4MnxxnKaNuVet_c,914
|
|
38
|
+
nshtrainer/config/_hf_hub/__init__.py,sha256=qNUK71CnviYcvd86Jjc7ZYFWjr6PLmYCynsVE7QmvQQ,1027
|
|
39
|
+
nshtrainer/config/callbacks/__init__.py,sha256=xEemxVCAaQA-iOnpp-ul-c3h_2DGjr4HGfwyavB01Sg,6491
|
|
40
|
+
nshtrainer/config/callbacks/actsave/__init__.py,sha256=ySlBq2LJbJgSsPtpT-raEiGSnZ3oFGTFB-MOCfZQlTM,773
|
|
41
|
+
nshtrainer/config/callbacks/base/__init__.py,sha256=HWeQLSg8wGFh9pFgw0eyRkfW2uO3KzM1Wgf-s6kmcQo,566
|
|
42
|
+
nshtrainer/config/callbacks/checkpoint/__init__.py,sha256=m1Xfmi16feKWry_dvz8oAS5Zlj-tT96EpY47eNmVMTQ,2664
|
|
43
|
+
nshtrainer/config/callbacks/checkpoint/_base/__init__.py,sha256=J3uc2rnMF2ZDioSIi3J5UrGRE52KA5zKoU6bR6M__TI,1219
|
|
44
|
+
nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=Ypgp3zt1W30zOvwGsCkrwq15KLe0ujlnJ_-MDlhM6i8,1601
|
|
45
|
+
nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=fL4sfS6BjIHBa4Kaaqmex6HSKw_y_maad78uzYiTVb0,1319
|
|
46
|
+
nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=L0cXPoWyZtOyo9ckyeggPuGanN_iCnHOXRpmN_bulxw,1033
|
|
47
|
+
nshtrainer/config/callbacks/debug_flag/__init__.py,sha256=Rplsl1OhADFF4v4PK_1tYhlPMzgfH70YAS8gmR_Ev4w,872
|
|
48
|
+
nshtrainer/config/callbacks/directory_setup/__init__.py,sha256=H8cckUCNuwmiIr_zYzh1robpRYRi2nNttl5IKaX6D1c,897
|
|
49
|
+
nshtrainer/config/callbacks/early_stopping/__init__.py,sha256=aths-c9sGApQEzikMSWLjBICpNz2mtCO8--DsW8ZjiM,1130
|
|
50
|
+
nshtrainer/config/callbacks/ema/__init__.py,sha256=xgAgcNBMAusIurhcV_W-95aW-g5jwPmNrtW9FwsW4YU,741
|
|
51
|
+
nshtrainer/config/callbacks/finite_checks/__init__.py,sha256=Ts4eCT26kuwQj0jE0VjZARuQjMB0hATE-KutM6PRmns,881
|
|
52
|
+
nshtrainer/config/callbacks/gradient_skipping/__init__.py,sha256=Hxk-feSH8H1T6nuTvSMt4yDOiSQ_h8Ob3dNlj3n6GM4,913
|
|
53
|
+
nshtrainer/config/callbacks/norm_logging/__init__.py,sha256=qnBnh1PTbomp9CpiniHwZs5BIW8Yi1YmLL-TDOA58Mk,856
|
|
54
|
+
nshtrainer/config/callbacks/print_table/__init__.py,sha256=rQDfNJHVpBNlG2en3ZQ-hCTN4iPl-7jIGi6owKOfkSc,893
|
|
55
|
+
nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py,sha256=0pb2FeArOZAQ0tNxjg1QwemsfoSim-jLDXFv0vlHGJI,909
|
|
56
|
+
nshtrainer/config/callbacks/shared_parameters/__init__.py,sha256=27EcGdNahNCWnu7pM62xelnfu4WMosrWrsudsNnVTjk,913
|
|
57
|
+
nshtrainer/config/callbacks/throughput_monitor/__init__.py,sha256=eui6qangrb3errf2jJ7slcY04aJ2sOBt8wyNDODeOlA,921
|
|
58
|
+
nshtrainer/config/callbacks/timer/__init__.py,sha256=3P-RroY0QGC_ayNbcQ7HqOUL8l7UcHIe9OJ416ve-h4,807
|
|
59
|
+
nshtrainer/config/callbacks/wandb_upload_code/__init__.py,sha256=XOFTN8ua7xHpXN63G1ZirqKTdGUs6i1yuwALFunxK-4,909
|
|
60
|
+
nshtrainer/config/callbacks/wandb_watch/__init__.py,sha256=TFykMCR_r40vyEL7enne96aE4Zy3YWaPcntpS26Giww,848
|
|
61
|
+
nshtrainer/config/loggers/__init__.py,sha256=R8veYxtV-dO-AeLCQ48oemUIxA3iNLw0OjpWafn9Dxo,2126
|
|
62
|
+
nshtrainer/config/loggers/_base/__init__.py,sha256=tD6WYqYmZ0qofaB7_XLIV347hwfd1XyrskAMVGyk0wA,526
|
|
63
|
+
nshtrainer/config/loggers/csv/__init__.py,sha256=sUTZG1pMNtaRzRLicGP7cYD-mWbfOWvhz492VOm0A9M,719
|
|
64
|
+
nshtrainer/config/loggers/tensorboard/__init__.py,sha256=O2WyxdWnQuKZLu-KooURC6k9Z4EAbplcjQc22boZY6k,860
|
|
65
|
+
nshtrainer/config/loggers/wandb/__init__.py,sha256=TGn6s1ofUEnO_jFBn8mdcjtDNc0avwDo1lJPbGUX6h0,1438
|
|
66
|
+
nshtrainer/config/lr_scheduler/__init__.py,sha256=fMUIOn7r4iIThhvqL0xGE-J5GVgbXQWUeprfpiYRHQY,2113
|
|
67
|
+
nshtrainer/config/lr_scheduler/_base/__init__.py,sha256=W68UxqYwSUHybd6rt7p2F0stNImbjUNN_aV8CDcXGhg,603
|
|
68
|
+
nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py,sha256=x1dl3U7YajRWNOkBmR8n4Q-P2dCrHTQ3tcTpfVi7jZg,1305
|
|
69
|
+
nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py,sha256=sRYCkXVcB0xQ3rOppigkNb7Hcz8-so_oLT8TUn8WBu0,1229
|
|
70
|
+
nshtrainer/config/metrics/__init__.py,sha256=MD3ltGcrXdYZmvllVGj1yoU3UOHsB-z46QGugEVDXiE,532
|
|
71
|
+
nshtrainer/config/metrics/_config/__init__.py,sha256=AJ-by6EzIkmBAxEzV5vgRUmVt-rsw2muPJBbkw8-BhY,514
|
|
72
|
+
nshtrainer/config/model/__init__.py,sha256=z0m6aZIUrx1B399-VpG_B-rWdJdqWXg7GlQ5SQiZaQw,1525
|
|
73
|
+
nshtrainer/config/model/base/__init__.py,sha256=Uot7t00zZpj4hZNay1AyvgKeClDYktRZqVyjnMmrxd8,699
|
|
74
|
+
nshtrainer/config/model/config/__init__.py,sha256=OIyHPEZtzxcKEZorchrBTxCXgvLAFKUraJW21JeF0mc,1495
|
|
75
|
+
nshtrainer/config/model/mixins/logger/__init__.py,sha256=kBWS3jdySOv77p7QuZtWMlw7rWdFWqodTeU6L8T9Q4Y,514
|
|
76
|
+
nshtrainer/config/nn/__init__.py,sha256=SW9ZIoqlXGRHEromIgUwlLlaAyk3thf0GPSTZvXGCtM,3954
|
|
77
|
+
nshtrainer/config/nn/mlp/__init__.py,sha256=Kqjklbj9ny7vGgf7EC4WA226JVauP5vdGK_orQzv4uk,898
|
|
78
|
+
nshtrainer/config/nn/nonlinearity/__init__.py,sha256=BIyxKR3S-1Fvf2JfwfkgZWu-6hr4nvjCa8ugHtjcQ0Q,4761
|
|
79
|
+
nshtrainer/config/optimizer/__init__.py,sha256=gOxCvg8Cl7FFSKZEekcIKRZMbpBZgXMiUFxOXrHWO4U,900
|
|
80
|
+
nshtrainer/config/profiler/__init__.py,sha256=OQb_MdbGni-hFHip_NrctnoIeG6Cks_8b2k3nPQh5XU,1485
|
|
81
|
+
nshtrainer/config/profiler/_base/__init__.py,sha256=zJ1dQIroQMmMkDYq-hAfLCahpAKAmfzI6eKf-SCzV2E,566
|
|
82
|
+
nshtrainer/config/profiler/advanced/__init__.py,sha256=53jR1agoCilDW8bKuT0NPgjQRwFy-GU38NFLnGkrpIU,856
|
|
83
|
+
nshtrainer/config/profiler/pytorch/__init__.py,sha256=LBxTukfv19-ISLg14r62a4YXERHnZiF1-QECfTtSD8M,848
|
|
84
|
+
nshtrainer/config/profiler/simple/__init__.py,sha256=dbRl6LWnr6aoIEf3sW_vwM1LAz2Nn3jrO0rltlayk4s,823
|
|
85
|
+
nshtrainer/config/runner/__init__.py,sha256=U5JWcg1u2oErwRnLwDiyArxDFutLWuordpDSaAJPtRs,488
|
|
86
|
+
nshtrainer/config/trainer/_config/__init__.py,sha256=7ZokJsepbCwP_rXOmvdEDxLYWhVkU8WnXQD4aJ5G6gs,6485
|
|
87
|
+
nshtrainer/config/trainer/checkpoint_connector/__init__.py,sha256=zR6sS2bZiaWnpQqixXQ9ki7qbMLxVqXozZpjXThcGjw,631
|
|
88
|
+
nshtrainer/config/util/_environment_info/__init__.py,sha256=hAX0-OtzahTKXkBURaPorREO4Wb-Ok_gsHHpFZ3p-GI,3694
|
|
89
|
+
nshtrainer/config/util/config/__init__.py,sha256=rQ_YrelDqwv9eYwHafgyZQOLA0fSV4EH7png-IkecdQ,1126
|
|
90
|
+
nshtrainer/config/util/config/dtype/__init__.py,sha256=Sik13kqmQvFn909ajIe4ISMKQgqn_P7D_Ce5S4o9Pig,514
|
|
91
|
+
nshtrainer/config/util/config/duration/__init__.py,sha256=KRAiveUvazReYX2U7JXMKyi8xIq8sbht50J8-Fabg0Q,1024
|
|
92
92
|
nshtrainer/data/__init__.py,sha256=fnigdEtNcB8DJqz4MHvcI6UwOZICFzWItskY78lLTMA,224
|
|
93
93
|
nshtrainer/data/balanced_batch_sampler.py,sha256=ybMJF-CguaZ17fLEweZ5suaGOiHOMEm3Bn8rQfGTzGQ,5445
|
|
94
94
|
nshtrainer/data/datamodule.py,sha256=Dk06LdcbK13sCLQWacdCoXU_CHWZxoKEZ_xpAg7nZQg,113
|
|
@@ -157,6 +157,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
|
|
|
157
157
|
nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
|
|
158
158
|
nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
|
|
159
159
|
nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
|
|
160
|
-
nshtrainer-0.
|
|
161
|
-
nshtrainer-0.
|
|
162
|
-
nshtrainer-0.
|
|
160
|
+
nshtrainer-0.42.0.dist-info/METADATA,sha256=KOHvIES5urRR9kMl8VrBaSNbz40XAl2cIpTaDMj08E8,916
|
|
161
|
+
nshtrainer-0.42.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
|
162
|
+
nshtrainer-0.42.0.dist-info/RECORD,,
|
|
File without changes
|