nshtrainer 0.40.4__py3-none-any.whl → 0.41.0__py3-none-any.whl

nshtrainer/config/lr_scheduler/__init__.py

@@ -0,0 +1,20 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.lr_scheduler import LRSchedulerConfigBase as LRSchedulerConfigBase
+from nshtrainer.lr_scheduler import LinearWarmupCosineDecayLRSchedulerConfig as LinearWarmupCosineDecayLRSchedulerConfig
+from nshtrainer.lr_scheduler import ReduceLROnPlateauConfig as ReduceLROnPlateauConfig
+from nshtrainer.lr_scheduler.reduce_lr_on_plateau import MetricConfig as MetricConfig
+
+# Type aliases
+from nshtrainer.lr_scheduler.linear_warmup_cosine import DurationConfig as DurationConfig
+from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
+
+# Submodule exports
+from . import _base as _base
+from . import linear_warmup_cosine as linear_warmup_cosine
+from . import reduce_lr_on_plateau as reduce_lr_on_plateau

nshtrainer/config/lr_scheduler/_base/__init__.py

@@ -0,0 +1,12 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.lr_scheduler._base import LRSchedulerConfigBase as LRSchedulerConfigBase
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py

@@ -0,0 +1,14 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.lr_scheduler.linear_warmup_cosine import LinearWarmupCosineDecayLRSchedulerConfig as LinearWarmupCosineDecayLRSchedulerConfig
+from nshtrainer.lr_scheduler.linear_warmup_cosine import LRSchedulerConfigBase as LRSchedulerConfigBase
+
+# Type aliases
+from nshtrainer.lr_scheduler.linear_warmup_cosine import DurationConfig as DurationConfig
+
+# Submodule exports

nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py

@@ -0,0 +1,14 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.lr_scheduler.reduce_lr_on_plateau import ReduceLROnPlateauConfig as ReduceLROnPlateauConfig
+from nshtrainer.lr_scheduler.reduce_lr_on_plateau import LRSchedulerConfigBase as LRSchedulerConfigBase
+from nshtrainer.lr_scheduler.reduce_lr_on_plateau import MetricConfig as MetricConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/metrics/__init__.py

@@ -0,0 +1,13 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.metrics import MetricConfig as MetricConfig
+
+# Type aliases
+
+# Submodule exports
+from . import _config as _config

nshtrainer/config/metrics/_config/__init__.py

@@ -0,0 +1,12 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.metrics._config import MetricConfig as MetricConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/model/__init__.py

@@ -0,0 +1,20 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.model import MetricConfig as MetricConfig
+from nshtrainer.model import BaseConfig as BaseConfig
+from nshtrainer.model import DirectoryConfig as DirectoryConfig
+from nshtrainer.model import TrainerConfig as TrainerConfig
+from nshtrainer.model.base import EnvironmentConfig as EnvironmentConfig
+from nshtrainer.model.config import CallbackConfigBase as CallbackConfigBase
+
+# Type aliases
+
+# Submodule exports
+from . import base as base
+from . import config as config
+from . import mixins as mixins

nshtrainer/config/model/base/__init__.py

@@ -0,0 +1,13 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.model.base import EnvironmentConfig as EnvironmentConfig
+from nshtrainer.model.base import BaseConfig as BaseConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/model/config/__init__.py

@@ -0,0 +1,17 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.model.config import MetricConfig as MetricConfig
+from nshtrainer.model.config import BaseConfig as BaseConfig
+from nshtrainer.model.config import DirectoryConfig as DirectoryConfig
+from nshtrainer.model.config import TrainerConfig as TrainerConfig
+from nshtrainer.model.config import EnvironmentConfig as EnvironmentConfig
+from nshtrainer.model.config import CallbackConfigBase as CallbackConfigBase
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/model/mixins/logger/__init__.py

@@ -0,0 +1,12 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.model.mixins.logger import BaseConfig as BaseConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/nn/__init__.py

@@ -0,0 +1,30 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.nn import MLPConfig as MLPConfig
+from nshtrainer.nn import BaseNonlinearityConfig as BaseNonlinearityConfig
+from nshtrainer.nn import TanhNonlinearityConfig as TanhNonlinearityConfig
+from nshtrainer.nn import ReLUNonlinearityConfig as ReLUNonlinearityConfig
+from nshtrainer.nn import PReLUConfig as PReLUConfig
+from nshtrainer.nn import SiLUNonlinearityConfig as SiLUNonlinearityConfig
+from nshtrainer.nn import LeakyReLUNonlinearityConfig as LeakyReLUNonlinearityConfig
+from nshtrainer.nn import SoftsignNonlinearityConfig as SoftsignNonlinearityConfig
+from nshtrainer.nn import MishNonlinearityConfig as MishNonlinearityConfig
+from nshtrainer.nn import SigmoidNonlinearityConfig as SigmoidNonlinearityConfig
+from nshtrainer.nn import SoftplusNonlinearityConfig as SoftplusNonlinearityConfig
+from nshtrainer.nn.nonlinearity import SwiGLUNonlinearityConfig as SwiGLUNonlinearityConfig
+from nshtrainer.nn import ELUNonlinearityConfig as ELUNonlinearityConfig
+from nshtrainer.nn import SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig
+from nshtrainer.nn import GELUNonlinearityConfig as GELUNonlinearityConfig
+from nshtrainer.nn import SwishNonlinearityConfig as SwishNonlinearityConfig
+
+# Type aliases
+from nshtrainer.nn import NonlinearityConfig as NonlinearityConfig
+
+# Submodule exports
+from . import mlp as mlp
+from . import nonlinearity as nonlinearity

nshtrainer/config/nn/mlp/__init__.py

@@ -0,0 +1,14 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.nn.mlp import MLPConfig as MLPConfig
+from nshtrainer.nn.mlp import BaseNonlinearityConfig as BaseNonlinearityConfig
+
+# Type aliases
+from nshtrainer.nn.mlp import NonlinearityConfig as NonlinearityConfig
+
+# Submodule exports

nshtrainer/config/nn/nonlinearity/__init__.py

@@ -0,0 +1,27 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.nn.nonlinearity import TanhNonlinearityConfig as TanhNonlinearityConfig
+from nshtrainer.nn.nonlinearity import ReLUNonlinearityConfig as ReLUNonlinearityConfig
+from nshtrainer.nn.nonlinearity import PReLUConfig as PReLUConfig
+from nshtrainer.nn.nonlinearity import SiLUNonlinearityConfig as SiLUNonlinearityConfig
+from nshtrainer.nn.nonlinearity import LeakyReLUNonlinearityConfig as LeakyReLUNonlinearityConfig
+from nshtrainer.nn.nonlinearity import SoftsignNonlinearityConfig as SoftsignNonlinearityConfig
+from nshtrainer.nn.nonlinearity import MishNonlinearityConfig as MishNonlinearityConfig
+from nshtrainer.nn.nonlinearity import SigmoidNonlinearityConfig as SigmoidNonlinearityConfig
+from nshtrainer.nn.nonlinearity import SoftplusNonlinearityConfig as SoftplusNonlinearityConfig
+from nshtrainer.nn.nonlinearity import SwiGLUNonlinearityConfig as SwiGLUNonlinearityConfig
+from nshtrainer.nn.nonlinearity import ELUNonlinearityConfig as ELUNonlinearityConfig
+from nshtrainer.nn.nonlinearity import BaseNonlinearityConfig as BaseNonlinearityConfig
+from nshtrainer.nn.nonlinearity import SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig
+from nshtrainer.nn.nonlinearity import GELUNonlinearityConfig as GELUNonlinearityConfig
+from nshtrainer.nn.nonlinearity import SwishNonlinearityConfig as SwishNonlinearityConfig
+
+# Type aliases
+from nshtrainer.nn.nonlinearity import NonlinearityConfig as NonlinearityConfig
+
+# Submodule exports

nshtrainer/config/optimizer/__init__.py

@@ -0,0 +1,14 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.optimizer import OptimizerConfigBase as OptimizerConfigBase
+from nshtrainer.optimizer import AdamWConfig as AdamWConfig
+
+# Type aliases
+from nshtrainer.optimizer import OptimizerConfig as OptimizerConfig
+
+# Submodule exports

nshtrainer/config/profiler/__init__.py

@@ -0,0 +1,20 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.profiler import BaseProfilerConfig as BaseProfilerConfig
+from nshtrainer.profiler import PyTorchProfilerConfig as PyTorchProfilerConfig
+from nshtrainer.profiler import AdvancedProfilerConfig as AdvancedProfilerConfig
+from nshtrainer.profiler import SimpleProfilerConfig as SimpleProfilerConfig
+
+# Type aliases
+from nshtrainer.profiler import ProfilerConfig as ProfilerConfig
+
+# Submodule exports
+from . import _base as _base
+from . import advanced as advanced
+from . import pytorch as pytorch
+from . import simple as simple

nshtrainer/config/profiler/_base/__init__.py

@@ -0,0 +1,12 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.profiler._base import BaseProfilerConfig as BaseProfilerConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/profiler/advanced/__init__.py

@@ -0,0 +1,13 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.profiler.advanced import AdvancedProfilerConfig as AdvancedProfilerConfig
+from nshtrainer.profiler.advanced import BaseProfilerConfig as BaseProfilerConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/profiler/pytorch/__init__.py

@@ -0,0 +1,13 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.profiler.pytorch import PyTorchProfilerConfig as PyTorchProfilerConfig
+from nshtrainer.profiler.pytorch import BaseProfilerConfig as BaseProfilerConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/profiler/simple/__init__.py

@@ -0,0 +1,13 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.profiler.simple import BaseProfilerConfig as BaseProfilerConfig
+from nshtrainer.profiler.simple import SimpleProfilerConfig as SimpleProfilerConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/runner/__init__.py

@@ -0,0 +1,12 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.runner import BaseConfig as BaseConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/trainer/_config/__init__.py

@@ -0,0 +1,35 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.trainer._config import BestCheckpointCallbackConfig as BestCheckpointCallbackConfig
+from nshtrainer.trainer._config import TensorboardLoggerConfig as TensorboardLoggerConfig
+from nshtrainer.trainer._config import CSVLoggerConfig as CSVLoggerConfig
+from nshtrainer.trainer._config import CheckpointSavingConfig as CheckpointSavingConfig
+from nshtrainer.trainer._config import SharedParametersConfig as SharedParametersConfig
+from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
+from nshtrainer.trainer._config import ReproducibilityConfig as ReproducibilityConfig
+from nshtrainer.trainer._config import HuggingFaceHubConfig as HuggingFaceHubConfig
+from nshtrainer.trainer._config import CheckpointLoadingConfig as CheckpointLoadingConfig
+from nshtrainer.trainer._config import WandbLoggerConfig as WandbLoggerConfig
+from nshtrainer.trainer._config import LoggingConfig as LoggingConfig
+from nshtrainer.trainer._config import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.trainer._config import LastCheckpointCallbackConfig as LastCheckpointCallbackConfig
+from nshtrainer.trainer._config import OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig
+from nshtrainer.trainer._config import RLPSanityChecksConfig as RLPSanityChecksConfig
+from nshtrainer.trainer._config import EarlyStoppingConfig as EarlyStoppingConfig
+from nshtrainer.trainer._config import OptimizationConfig as OptimizationConfig
+from nshtrainer.trainer._config import TrainerConfig as TrainerConfig
+from nshtrainer.trainer._config import DebugFlagCallbackConfig as DebugFlagCallbackConfig
+from nshtrainer.trainer._config import GradientClippingConfig as GradientClippingConfig
+
+# Type aliases
+from nshtrainer.trainer._config import CallbackConfig as CallbackConfig
+from nshtrainer.trainer._config import CheckpointCallbackConfig as CheckpointCallbackConfig
+from nshtrainer.trainer._config import LoggerConfig as LoggerConfig
+from nshtrainer.trainer._config import ProfilerConfig as ProfilerConfig
+
+# Submodule exports

nshtrainer/config/trainer/checkpoint_connector/__init__.py

@@ -0,0 +1,12 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.trainer.checkpoint_connector import CheckpointLoadingConfig as CheckpointLoadingConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/util/_environment_info/__init__.py

@@ -0,0 +1,22 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.util._environment_info import EnvironmentSnapshotConfig as EnvironmentSnapshotConfig
+from nshtrainer.util._environment_info import EnvironmentHardwareConfig as EnvironmentHardwareConfig
+from nshtrainer.util._environment_info import EnvironmentPackageConfig as EnvironmentPackageConfig
+from nshtrainer.util._environment_info import GitRepositoryConfig as GitRepositoryConfig
+from nshtrainer.util._environment_info import EnvironmentGPUConfig as EnvironmentGPUConfig
+from nshtrainer.util._environment_info import EnvironmentCUDAConfig as EnvironmentCUDAConfig
+from nshtrainer.util._environment_info import EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig
+from nshtrainer.util._environment_info import EnvironmentClassInformationConfig as EnvironmentClassInformationConfig
+from nshtrainer.util._environment_info import EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig
+from nshtrainer.util._environment_info import EnvironmentConfig as EnvironmentConfig
+from nshtrainer.util._environment_info import EnvironmentLSFInformationConfig as EnvironmentLSFInformationConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/util/config/__init__.py

@@ -0,0 +1,17 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.util.config import EpochsConfig as EpochsConfig
+from nshtrainer.util.config import StepsConfig as StepsConfig
+from nshtrainer.util.config import DTypeConfig as DTypeConfig
+
+# Type aliases
+from nshtrainer.util.config import DurationConfig as DurationConfig
+
+# Submodule exports
+from . import dtype as dtype
+from . import duration as duration

nshtrainer/config/util/config/dtype/__init__.py

@@ -0,0 +1,12 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.util.config.dtype import DTypeConfig as DTypeConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/util/config/duration/__init__.py

@@ -0,0 +1,14 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.util.config.duration import EpochsConfig as EpochsConfig
+from nshtrainer.util.config.duration import StepsConfig as StepsConfig
+
+# Type aliases
+from nshtrainer.util.config.duration import DurationConfig as DurationConfig
+
+# Submodule exports

{nshtrainer-0.40.4.dist-info → nshtrainer-0.41.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshtrainer
-Version: 0.40.4
+Version: 0.41.0
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshtrainer-0.40.4.dist-info → nshtrainer-0.41.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-nshtrainer/__init__.py,sha256=8hx3uqzroMYOQM7S4U3Eznw7MHF3YDqUkO1MDQeC4cM,642
+nshtrainer/__init__.py,sha256=7_l5lxWMqS661UBeT3XZ8-RLaiQJhoRvLgOtXglBEhU,645
 nshtrainer/_callback.py,sha256=A1zLsTy4b_wOYnInLLXGSRdHzT2yNa6mPEql-ozm0u0,1013
 nshtrainer/_checkpoint/loader.py,sha256=5vjg-OFChXJjgiOVv8vnV8nwTscfdDtEdxQRz6uPfDE,14158
 nshtrainer/_checkpoint/metadata.py,sha256=5D4PgKodzhLsmQvuF3xxkH49epKaegxi4wh_ImDTtns,4737
@@ -31,7 +31,64 @@ nshtrainer/callbacks/throughput_monitor.py,sha256=H_ocXErZxUO3dxFk8Tx_VQdpI9E_Zt
 nshtrainer/callbacks/timer.py,sha256=quS79oYClDUvQxJkNWmDMe0hwRUkkREgTgqzVrnom50,4607
 nshtrainer/callbacks/wandb_upload_code.py,sha256=OWG4UkL2SfW6oj6AGRXeBJsZmgsqeHLW2Fj8Jm4ga3I,2298
 nshtrainer/callbacks/wandb_watch.py,sha256=Y6SEXfIx3kDDQbI5zpP53BVq0FBLJbLd3RJsiHZk1-Y,2921
-nshtrainer/config.py,sha256=pZyRZOkBRR7eBFRiHpHjQFNEFjaX9tYZIAqZtvKi6cA,8312
+nshtrainer/config/__init__.py,sha256=y-K4iv44DZkpV0cIQglNLmxHbxVBVRO5DLq9E34M-HI,7808
+nshtrainer/config/_checkpoint/loader/__init__.py,sha256=wpbjL4GsMfYmb0PM3yj6Qry-d7aKxM4yZ7s7Qp9Qn3g,803
+nshtrainer/config/_checkpoint/metadata/__init__.py,sha256=3iSkHEmKTw0nCaGfzWroUHN6h6naHfLOi7EySCH8yKw,282
+nshtrainer/config/_directory/__init__.py,sha256=1Qf7Q0OOwwCQmPNm1VM7vFjuaL25B-Wnzcz5ZtuSESI,325
+nshtrainer/config/_hf_hub/__init__.py,sha256=7qz4dzkjoOocv9biRTKCVsQI-GgSE5dAOft2s7tGgsw,358
+nshtrainer/config/callbacks/__init__.py,sha256=u9bcllwXYEGxSWFKcDDP16qQzPx_P2eGYqKDRdKMc-c,2698
+nshtrainer/config/callbacks/actsave/__init__.py,sha256=LB2dnKpCUTy2_15hpQgZ77Y_9GmALlxK2V01CpyCIQ8,268
+nshtrainer/config/callbacks/base/__init__.py,sha256=V4fzP1q_LpXPoGV5Ve5mXFRscNJ0AQkr-4k5iJDWBxU,193
+nshtrainer/config/callbacks/checkpoint/__init__.py,sha256=d3Y0I3aomhvyVva6PF5nraJb3aM9rb5gDZNB4_7z7mg,1017
+nshtrainer/config/callbacks/checkpoint/_base/__init__.py,sha256=7iZ3aTak0NOiLI5s5tOpY-PJyq6h8wEybf0U0VeuBNk,407
+nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=rST4HhhOhti10QZjWBu0CkJErjNTI8cP44e0hQtX_j8,546
+nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=gMxD2quXg7VCa1vkzjSSjegtBXgQJ6lpvq-OaLeicFI,457
+nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=2WHicpnirFzydPA9wsRDXqN6xwT2W7eVqCdDNqtX9xs,366
+nshtrainer/config/callbacks/debug_flag/__init__.py,sha256=hiiZvFDLjUDpH3Xpk9IerIRNTmHhycJVSEMPPrrqUi8,294
+nshtrainer/config/callbacks/directory_setup/__init__.py,sha256=NswK3ywE-oikMUzYgJ3-bXK1TPcf5OB5Ce2mslO4gLY,298
+nshtrainer/config/callbacks/early_stopping/__init__.py,sha256=VpkNZ7z0s_iQP_IGsUNk-s6c-hSiqLydHFQ1NkqCjjM,371
+nshtrainer/config/callbacks/ema/__init__.py,sha256=id-m1X5Q-Rkw9HJo4J72V59EjedOxlHiqLW4GWa8J34,252
+nshtrainer/config/callbacks/finite_checks/__init__.py,sha256=utfprFQgrhrk3CFaN6twRu35ryJzD_L2qwFYN5VmJcM,290
+nshtrainer/config/callbacks/gradient_skipping/__init__.py,sha256=mrPYNjVQMgKfNM8rStmNAeATOTus1685D8LB4JYbszs,306
+nshtrainer/config/callbacks/norm_logging/__init__.py,sha256=w2l08YgAbxy3EjDaZO3f0s8NgURXFZnW5xy1n8JFUKU,286
+nshtrainer/config/callbacks/print_table/__init__.py,sha256=Ne6O1JbSnPwCj4H225FBbJnHRoIR4rBnIG2pogLeHZU,296
+nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py,sha256=-VVBo8P7JgI0J2JbehOqi4BnrDxwmRrBkZBxQSsDNKE,304
+nshtrainer/config/callbacks/shared_parameters/__init__.py,sha256=UH6U4_xaBwpk-otAi2k4un9sNNDOykJ0l26vkq4tX1c,306
+nshtrainer/config/callbacks/throughput_monitor/__init__.py,sha256=Eafoki1Ja4qEgiGmCxLVimQtB5ObxVcqRDfdYEjoTX0,310
+nshtrainer/config/callbacks/timer/__init__.py,sha256=pl0ZiZZdeJI9im65mkVQ1NoYP_ZgV-vHGXLtecjLAXM,270
+nshtrainer/config/callbacks/wandb_upload_code/__init__.py,sha256=QGwSTtQZFMJnVIGZsMwbAMRHy_qvHB-S4ysEvT47VGU,304
+nshtrainer/config/callbacks/wandb_watch/__init__.py,sha256=1qdlXUkzfuLCr6_x9Yor1rnIrxai2Ev5swSl3T0Tjfo,282
+nshtrainer/config/loggers/__init__.py,sha256=c5bs11U2DCHUeqrQsPPzhZVOu_OHoZrFVXdLsewOMcY,820
+nshtrainer/config/loggers/_base/__init__.py,sha256=EAQOKsfk-c9hIPZ6ohjv1r9WpDCVD5mWVGrfu5ST5OY,188
+nshtrainer/config/loggers/csv/__init__.py,sha256=YnUMUPucEHaBSiXErhAlP_GOTckb_g530ttoCy-rJdk,256
+nshtrainer/config/loggers/tensorboard/__init__.py,sha256=nq18V6CMd-kz7mvgxEgNUqztNrU8dbvdNbwZBGJcwe4,288
+nshtrainer/config/loggers/wandb/__init__.py,sha256=nY6plVAb0f3NBenq3LYHnXy_XauqGQWnGw7Z8ik_Ug0,500
+nshtrainer/config/lr_scheduler/__init__.py,sha256=VNAFT6BbQQ4EPY7AfCMTCba0-K5wr0cp5c4FwK67Aj4,803
+nshtrainer/config/lr_scheduler/_base/__init__.py,sha256=RO6gI-P9SOgH17m3ycRns9IgiFda6oIOAV4Xqa9-0og,203
+nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py,sha256=uzisQaTWTVxZL3mPQJBdJy16hrfiCUAzP4hgRdWDcow,450
+nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py,sha256=yI7Hpa4CNX4Ys7QIq_emLiWbr6tghf-oVjVjIq5k1hE,412
+nshtrainer/config/metrics/__init__.py,sha256=P6goxe9RhwQZx8cC2gZ1dfoofLOTII-O-Y1Gas3km9k,207
+nshtrainer/config/metrics/_config/__init__.py,sha256=lTdHS7SFD-8VFkalFqNltPrhWxbWFEovt1XPmC8cusc,182
+nshtrainer/config/model/__init__.py,sha256=u8pWp7RaDqG9kAlKVnMFzOCNyqisOmk5ouH1FQ_xY-4,589
+nshtrainer/config/model/base/__init__.py,sha256=zL1ZUA8OaeGSyA1Beyf_UmNjEWHayHdX5KNxFCkxMcg,246
+nshtrainer/config/model/config/__init__.py,sha256=rD2CeNx3prtnxv5xNluMiDbSotRpyJnHZhNxVG5vuaM,530
+nshtrainer/config/model/mixins/logger/__init__.py,sha256=phExLK3R5R5okUA-DcekxoWZgU1oEgIyzdOG9jL3TUY,182
+nshtrainer/config/nn/__init__.py,sha256=AutesBnPzVh1CWXe8yYXwr_ZxQ4KGdUOAd1BTp1tjw0,1456
+nshtrainer/config/nn/mlp/__init__.py,sha256=F5eSCQM8lqTA7fGDFeQLX1vyS0BAg6Br1SFbwI3z2E4,317
+nshtrainer/config/nn/nonlinearity/__init__.py,sha256=lhx-Mitn6BZ-wdDuE70OT9mWNycPs_CoWLDmuLQxZdo,1534
+nshtrainer/config/optimizer/__init__.py,sha256=LxnWMttr594voY2iewD-QgpLJ0LYQG2h3fCxiJPFd5o,318
+nshtrainer/config/profiler/__init__.py,sha256=cKvQwqdrc4htiYsL5ahAFdoJw0IGqnK_Nf99suDYstw,617
+nshtrainer/config/profiler/_base/__init__.py,sha256=Pi5HckHRSywRsvn9Cd84tqRwZD5UH8cDOS4_f8zM8P0,193
+nshtrainer/config/profiler/advanced/__init__.py,sha256=gXn8K98CUCYy4m_Efv0yIuOnvxBvKdmZc_7r6tmsKec,286
+nshtrainer/config/profiler/pytorch/__init__.py,sha256=UO_FQhue8hpHS7_WsdMDw_dWTtDi_gFvnr1TBWk2u7U,282
+nshtrainer/config/profiler/simple/__init__.py,sha256=Wbec--_xC1oUkmiA7pZw4f5emuMTEOn-n6vJ3Jk0Jas,278
+nshtrainer/config/runner/__init__.py,sha256=NAvbAyrdKy8u_uRaNCHP7eJz9X2anXQAB75TPYyDbe0,169
+nshtrainer/config/trainer/_config/__init__.py,sha256=MG9d1ayQWlV54xoBRQlBUHxG2ihdBqmH1b0VtdFpdSk,2140
+nshtrainer/config/trainer/checkpoint_connector/__init__.py,sha256=CM2wmQ8KirlMqx4EYnttL_M4fjdT9MhYQjvhqZRyyRM,217
+nshtrainer/config/util/_environment_info/__init__.py,sha256=e11YIxaFL_6o7CiYIn80mYfaPSUmvKJ2RfftAhjTzeg,1237
+nshtrainer/config/util/config/__init__.py,sha256=5PQPUvyx_ycS-H6NdGebFS2W_6cFozLsrjeI-ITluVQ,434
+nshtrainer/config/util/config/dtype/__init__.py,sha256=fzOWOq-WMhNw-ZOoU3VnaPAKPvVy4IfL2L0MEzdTRg0,182
+nshtrainer/config/util/config/duration/__init__.py,sha256=zOA-9xum4WhS1IWfP1-YP__u1Axzc5js-9Xz015QFQ0,335
 nshtrainer/data/__init__.py,sha256=fnigdEtNcB8DJqz4MHvcI6UwOZICFzWItskY78lLTMA,224
 nshtrainer/data/balanced_batch_sampler.py,sha256=ybMJF-CguaZ17fLEweZ5suaGOiHOMEm3Bn8rQfGTzGQ,5445
 nshtrainer/data/datamodule.py,sha256=Dk06LdcbK13sCLQWacdCoXU_CHWZxoKEZ_xpAg7nZQg,113
@@ -100,6 +157,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
 nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
 nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
 nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
-nshtrainer-0.40.4.dist-info/METADATA,sha256=WEVuvm57GaDclfY6NGCfcUuyiaGyF8iljtBvy4uyUX8,916
-nshtrainer-0.40.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-0.40.4.dist-info/RECORD,,
+nshtrainer-0.41.0.dist-info/METADATA,sha256=uN5z13Tg8ahZJokfHavELvxMSpmVVvWbdIDOIZ6kQI8,916
+nshtrainer-0.41.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-0.41.0.dist-info/RECORD,,