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.
Files changed (62) hide show
  1. nshtrainer/__init__.py +1 -2
  2. nshtrainer/config/__init__.py +406 -95
  3. nshtrainer/config/_checkpoint/loader/__init__.py +55 -13
  4. nshtrainer/config/_checkpoint/metadata/__init__.py +22 -8
  5. nshtrainer/config/_directory/__init__.py +21 -9
  6. nshtrainer/config/_hf_hub/__init__.py +25 -9
  7. nshtrainer/config/callbacks/__init__.py +114 -29
  8. nshtrainer/config/callbacks/actsave/__init__.py +20 -8
  9. nshtrainer/config/callbacks/base/__init__.py +17 -7
  10. nshtrainer/config/callbacks/checkpoint/__init__.py +62 -13
  11. nshtrainer/config/callbacks/checkpoint/_base/__init__.py +33 -9
  12. nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py +40 -10
  13. nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py +33 -9
  14. nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py +26 -8
  15. nshtrainer/config/callbacks/debug_flag/__init__.py +24 -8
  16. nshtrainer/config/callbacks/directory_setup/__init__.py +26 -8
  17. nshtrainer/config/callbacks/early_stopping/__init__.py +31 -9
  18. nshtrainer/config/callbacks/ema/__init__.py +20 -8
  19. nshtrainer/config/callbacks/finite_checks/__init__.py +26 -8
  20. nshtrainer/config/callbacks/gradient_skipping/__init__.py +26 -8
  21. nshtrainer/config/callbacks/norm_logging/__init__.py +24 -8
  22. nshtrainer/config/callbacks/print_table/__init__.py +26 -8
  23. nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py +26 -8
  24. nshtrainer/config/callbacks/shared_parameters/__init__.py +26 -8
  25. nshtrainer/config/callbacks/throughput_monitor/__init__.py +26 -8
  26. nshtrainer/config/callbacks/timer/__init__.py +22 -8
  27. nshtrainer/config/callbacks/wandb_upload_code/__init__.py +26 -8
  28. nshtrainer/config/callbacks/wandb_watch/__init__.py +24 -8
  29. nshtrainer/config/loggers/__init__.py +41 -14
  30. nshtrainer/config/loggers/_base/__init__.py +15 -7
  31. nshtrainer/config/loggers/csv/__init__.py +18 -8
  32. nshtrainer/config/loggers/tensorboard/__init__.py +24 -8
  33. nshtrainer/config/loggers/wandb/__init__.py +31 -11
  34. nshtrainer/config/lr_scheduler/__init__.py +49 -12
  35. nshtrainer/config/lr_scheduler/_base/__init__.py +19 -7
  36. nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py +33 -9
  37. nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py +33 -9
  38. nshtrainer/config/metrics/__init__.py +16 -7
  39. nshtrainer/config/metrics/_config/__init__.py +15 -7
  40. nshtrainer/config/model/__init__.py +31 -12
  41. nshtrainer/config/model/base/__init__.py +18 -8
  42. nshtrainer/config/model/config/__init__.py +30 -12
  43. nshtrainer/config/model/mixins/logger/__init__.py +15 -7
  44. nshtrainer/config/nn/__init__.py +68 -23
  45. nshtrainer/config/nn/mlp/__init__.py +21 -9
  46. nshtrainer/config/nn/nonlinearity/__init__.py +118 -22
  47. nshtrainer/config/optimizer/__init__.py +21 -9
  48. nshtrainer/config/profiler/__init__.py +28 -11
  49. nshtrainer/config/profiler/_base/__init__.py +17 -7
  50. nshtrainer/config/profiler/advanced/__init__.py +24 -8
  51. nshtrainer/config/profiler/pytorch/__init__.py +24 -8
  52. nshtrainer/config/profiler/simple/__init__.py +22 -8
  53. nshtrainer/config/runner/__init__.py +15 -7
  54. nshtrainer/config/trainer/_config/__init__.py +144 -30
  55. nshtrainer/config/trainer/checkpoint_connector/__init__.py +19 -7
  56. nshtrainer/config/util/_environment_info/__init__.py +87 -17
  57. nshtrainer/config/util/config/__init__.py +25 -10
  58. nshtrainer/config/util/config/dtype/__init__.py +15 -7
  59. nshtrainer/config/util/config/duration/__init__.py +27 -9
  60. {nshtrainer-0.41.0.dist-info → nshtrainer-0.42.0.dist-info}/METADATA +1 -1
  61. {nshtrainer-0.41.0.dist-info → nshtrainer-0.42.0.dist-info}/RECORD +62 -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
- # Config classes
8
- from nshtrainer.util._environment_info import EnvironmentSnapshotConfig as EnvironmentSnapshotConfig
9
- from nshtrainer.util._environment_info import EnvironmentHardwareConfig as EnvironmentHardwareConfig
10
- from nshtrainer.util._environment_info import EnvironmentPackageConfig as EnvironmentPackageConfig
11
- from nshtrainer.util._environment_info import GitRepositoryConfig as GitRepositoryConfig
12
- from nshtrainer.util._environment_info import EnvironmentGPUConfig as EnvironmentGPUConfig
13
- from nshtrainer.util._environment_info import EnvironmentCUDAConfig as EnvironmentCUDAConfig
14
- from nshtrainer.util._environment_info import EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig
15
- from nshtrainer.util._environment_info import EnvironmentClassInformationConfig as EnvironmentClassInformationConfig
16
- from nshtrainer.util._environment_info import EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig
17
- from nshtrainer.util._environment_info import EnvironmentConfig as EnvironmentConfig
18
- from nshtrainer.util._environment_info import EnvironmentLSFInformationConfig as EnvironmentLSFInformationConfig
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
- # Type aliases
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
- # Config classes
8
- from nshtrainer.util.config import EpochsConfig as EpochsConfig
9
- from nshtrainer.util.config import StepsConfig as StepsConfig
10
- from nshtrainer.util.config import DTypeConfig as DTypeConfig
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
- # Config classes
8
- from nshtrainer.util.config.dtype import DTypeConfig as DTypeConfig
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
- # Type aliases
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
- # Config classes
8
- from nshtrainer.util.config.duration import EpochsConfig as EpochsConfig
9
- from nshtrainer.util.config.duration import StepsConfig as StepsConfig
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
- # Type aliases
12
- from nshtrainer.util.config.duration import DurationConfig as DurationConfig
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,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.41.0
3
+ Version: 0.42.0
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
@@ -1,4 +1,4 @@
1
- nshtrainer/__init__.py,sha256=7_l5lxWMqS661UBeT3XZ8-RLaiQJhoRvLgOtXglBEhU,645
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=y-K4iv44DZkpV0cIQglNLmxHbxVBVRO5DLq9E34M-HI,7808
35
- nshtrainer/config/_checkpoint/loader/__init__.py,sha256=wpbjL4GsMfYmb0PM3yj6Qry-d7aKxM4yZ7s7Qp9Qn3g,803
36
- nshtrainer/config/_checkpoint/metadata/__init__.py,sha256=3iSkHEmKTw0nCaGfzWroUHN6h6naHfLOi7EySCH8yKw,282
37
- nshtrainer/config/_directory/__init__.py,sha256=1Qf7Q0OOwwCQmPNm1VM7vFjuaL25B-Wnzcz5ZtuSESI,325
38
- nshtrainer/config/_hf_hub/__init__.py,sha256=7qz4dzkjoOocv9biRTKCVsQI-GgSE5dAOft2s7tGgsw,358
39
- nshtrainer/config/callbacks/__init__.py,sha256=u9bcllwXYEGxSWFKcDDP16qQzPx_P2eGYqKDRdKMc-c,2698
40
- nshtrainer/config/callbacks/actsave/__init__.py,sha256=LB2dnKpCUTy2_15hpQgZ77Y_9GmALlxK2V01CpyCIQ8,268
41
- nshtrainer/config/callbacks/base/__init__.py,sha256=V4fzP1q_LpXPoGV5Ve5mXFRscNJ0AQkr-4k5iJDWBxU,193
42
- nshtrainer/config/callbacks/checkpoint/__init__.py,sha256=d3Y0I3aomhvyVva6PF5nraJb3aM9rb5gDZNB4_7z7mg,1017
43
- nshtrainer/config/callbacks/checkpoint/_base/__init__.py,sha256=7iZ3aTak0NOiLI5s5tOpY-PJyq6h8wEybf0U0VeuBNk,407
44
- nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=rST4HhhOhti10QZjWBu0CkJErjNTI8cP44e0hQtX_j8,546
45
- nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=gMxD2quXg7VCa1vkzjSSjegtBXgQJ6lpvq-OaLeicFI,457
46
- nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=2WHicpnirFzydPA9wsRDXqN6xwT2W7eVqCdDNqtX9xs,366
47
- nshtrainer/config/callbacks/debug_flag/__init__.py,sha256=hiiZvFDLjUDpH3Xpk9IerIRNTmHhycJVSEMPPrrqUi8,294
48
- nshtrainer/config/callbacks/directory_setup/__init__.py,sha256=NswK3ywE-oikMUzYgJ3-bXK1TPcf5OB5Ce2mslO4gLY,298
49
- nshtrainer/config/callbacks/early_stopping/__init__.py,sha256=VpkNZ7z0s_iQP_IGsUNk-s6c-hSiqLydHFQ1NkqCjjM,371
50
- nshtrainer/config/callbacks/ema/__init__.py,sha256=id-m1X5Q-Rkw9HJo4J72V59EjedOxlHiqLW4GWa8J34,252
51
- nshtrainer/config/callbacks/finite_checks/__init__.py,sha256=utfprFQgrhrk3CFaN6twRu35ryJzD_L2qwFYN5VmJcM,290
52
- nshtrainer/config/callbacks/gradient_skipping/__init__.py,sha256=mrPYNjVQMgKfNM8rStmNAeATOTus1685D8LB4JYbszs,306
53
- nshtrainer/config/callbacks/norm_logging/__init__.py,sha256=w2l08YgAbxy3EjDaZO3f0s8NgURXFZnW5xy1n8JFUKU,286
54
- nshtrainer/config/callbacks/print_table/__init__.py,sha256=Ne6O1JbSnPwCj4H225FBbJnHRoIR4rBnIG2pogLeHZU,296
55
- nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py,sha256=-VVBo8P7JgI0J2JbehOqi4BnrDxwmRrBkZBxQSsDNKE,304
56
- nshtrainer/config/callbacks/shared_parameters/__init__.py,sha256=UH6U4_xaBwpk-otAi2k4un9sNNDOykJ0l26vkq4tX1c,306
57
- nshtrainer/config/callbacks/throughput_monitor/__init__.py,sha256=Eafoki1Ja4qEgiGmCxLVimQtB5ObxVcqRDfdYEjoTX0,310
58
- nshtrainer/config/callbacks/timer/__init__.py,sha256=pl0ZiZZdeJI9im65mkVQ1NoYP_ZgV-vHGXLtecjLAXM,270
59
- nshtrainer/config/callbacks/wandb_upload_code/__init__.py,sha256=QGwSTtQZFMJnVIGZsMwbAMRHy_qvHB-S4ysEvT47VGU,304
60
- nshtrainer/config/callbacks/wandb_watch/__init__.py,sha256=1qdlXUkzfuLCr6_x9Yor1rnIrxai2Ev5swSl3T0Tjfo,282
61
- nshtrainer/config/loggers/__init__.py,sha256=c5bs11U2DCHUeqrQsPPzhZVOu_OHoZrFVXdLsewOMcY,820
62
- nshtrainer/config/loggers/_base/__init__.py,sha256=EAQOKsfk-c9hIPZ6ohjv1r9WpDCVD5mWVGrfu5ST5OY,188
63
- nshtrainer/config/loggers/csv/__init__.py,sha256=YnUMUPucEHaBSiXErhAlP_GOTckb_g530ttoCy-rJdk,256
64
- nshtrainer/config/loggers/tensorboard/__init__.py,sha256=nq18V6CMd-kz7mvgxEgNUqztNrU8dbvdNbwZBGJcwe4,288
65
- nshtrainer/config/loggers/wandb/__init__.py,sha256=nY6plVAb0f3NBenq3LYHnXy_XauqGQWnGw7Z8ik_Ug0,500
66
- nshtrainer/config/lr_scheduler/__init__.py,sha256=VNAFT6BbQQ4EPY7AfCMTCba0-K5wr0cp5c4FwK67Aj4,803
67
- nshtrainer/config/lr_scheduler/_base/__init__.py,sha256=RO6gI-P9SOgH17m3ycRns9IgiFda6oIOAV4Xqa9-0og,203
68
- nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py,sha256=uzisQaTWTVxZL3mPQJBdJy16hrfiCUAzP4hgRdWDcow,450
69
- nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py,sha256=yI7Hpa4CNX4Ys7QIq_emLiWbr6tghf-oVjVjIq5k1hE,412
70
- nshtrainer/config/metrics/__init__.py,sha256=P6goxe9RhwQZx8cC2gZ1dfoofLOTII-O-Y1Gas3km9k,207
71
- nshtrainer/config/metrics/_config/__init__.py,sha256=lTdHS7SFD-8VFkalFqNltPrhWxbWFEovt1XPmC8cusc,182
72
- nshtrainer/config/model/__init__.py,sha256=u8pWp7RaDqG9kAlKVnMFzOCNyqisOmk5ouH1FQ_xY-4,589
73
- nshtrainer/config/model/base/__init__.py,sha256=zL1ZUA8OaeGSyA1Beyf_UmNjEWHayHdX5KNxFCkxMcg,246
74
- nshtrainer/config/model/config/__init__.py,sha256=rD2CeNx3prtnxv5xNluMiDbSotRpyJnHZhNxVG5vuaM,530
75
- nshtrainer/config/model/mixins/logger/__init__.py,sha256=phExLK3R5R5okUA-DcekxoWZgU1oEgIyzdOG9jL3TUY,182
76
- nshtrainer/config/nn/__init__.py,sha256=AutesBnPzVh1CWXe8yYXwr_ZxQ4KGdUOAd1BTp1tjw0,1456
77
- nshtrainer/config/nn/mlp/__init__.py,sha256=F5eSCQM8lqTA7fGDFeQLX1vyS0BAg6Br1SFbwI3z2E4,317
78
- nshtrainer/config/nn/nonlinearity/__init__.py,sha256=lhx-Mitn6BZ-wdDuE70OT9mWNycPs_CoWLDmuLQxZdo,1534
79
- nshtrainer/config/optimizer/__init__.py,sha256=LxnWMttr594voY2iewD-QgpLJ0LYQG2h3fCxiJPFd5o,318
80
- nshtrainer/config/profiler/__init__.py,sha256=cKvQwqdrc4htiYsL5ahAFdoJw0IGqnK_Nf99suDYstw,617
81
- nshtrainer/config/profiler/_base/__init__.py,sha256=Pi5HckHRSywRsvn9Cd84tqRwZD5UH8cDOS4_f8zM8P0,193
82
- nshtrainer/config/profiler/advanced/__init__.py,sha256=gXn8K98CUCYy4m_Efv0yIuOnvxBvKdmZc_7r6tmsKec,286
83
- nshtrainer/config/profiler/pytorch/__init__.py,sha256=UO_FQhue8hpHS7_WsdMDw_dWTtDi_gFvnr1TBWk2u7U,282
84
- nshtrainer/config/profiler/simple/__init__.py,sha256=Wbec--_xC1oUkmiA7pZw4f5emuMTEOn-n6vJ3Jk0Jas,278
85
- nshtrainer/config/runner/__init__.py,sha256=NAvbAyrdKy8u_uRaNCHP7eJz9X2anXQAB75TPYyDbe0,169
86
- nshtrainer/config/trainer/_config/__init__.py,sha256=MG9d1ayQWlV54xoBRQlBUHxG2ihdBqmH1b0VtdFpdSk,2140
87
- nshtrainer/config/trainer/checkpoint_connector/__init__.py,sha256=CM2wmQ8KirlMqx4EYnttL_M4fjdT9MhYQjvhqZRyyRM,217
88
- nshtrainer/config/util/_environment_info/__init__.py,sha256=e11YIxaFL_6o7CiYIn80mYfaPSUmvKJ2RfftAhjTzeg,1237
89
- nshtrainer/config/util/config/__init__.py,sha256=5PQPUvyx_ycS-H6NdGebFS2W_6cFozLsrjeI-ITluVQ,434
90
- nshtrainer/config/util/config/dtype/__init__.py,sha256=fzOWOq-WMhNw-ZOoU3VnaPAKPvVy4IfL2L0MEzdTRg0,182
91
- nshtrainer/config/util/config/duration/__init__.py,sha256=zOA-9xum4WhS1IWfP1-YP__u1Axzc5js-9Xz015QFQ0,335
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.41.0.dist-info/METADATA,sha256=uN5z13Tg8ahZJokfHavELvxMSpmVVvWbdIDOIZ6kQI8,916
161
- nshtrainer-0.41.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
162
- nshtrainer-0.41.0.dist-info/RECORD,,
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,,