nshtrainer 0.44.1__py3-none-any.whl → 1.0.0b10__py3-none-any.whl

nshtrainer/config/loggers/__init__.py

@@ -7,6 +7,7 @@ from typing import TYPE_CHECKING
 # Config/alias imports
 
 if TYPE_CHECKING:
+    from nshtrainer.loggers import ActSaveLoggerConfig as ActSaveLoggerConfig
     from nshtrainer.loggers import BaseLoggerConfig as BaseLoggerConfig
     from nshtrainer.loggers import CSVLoggerConfig as CSVLoggerConfig
     from nshtrainer.loggers import LoggerConfig as LoggerConfig
@@ -26,8 +27,16 @@ else:
 
         if name in globals():
             return globals()[name]
+        if name == "ActSaveLoggerConfig":
+            return importlib.import_module("nshtrainer.loggers").ActSaveLoggerConfig
         if name == "BaseLoggerConfig":
             return importlib.import_module("nshtrainer.loggers").BaseLoggerConfig
+        if name == "CSVLoggerConfig":
+            return importlib.import_module("nshtrainer.loggers").CSVLoggerConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.loggers.wandb"
+            ).CallbackConfigBase
         if name == "TensorboardLoggerConfig":
             return importlib.import_module("nshtrainer.loggers").TensorboardLoggerConfig
         if name == "WandbLoggerConfig":
@@ -40,12 +49,6 @@ else:
             return importlib.import_module(
                 "nshtrainer.loggers.wandb"
             ).WandbWatchCallbackConfig
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.loggers.wandb"
-            ).CallbackConfigBase
-        if name == "CSVLoggerConfig":
-            return importlib.import_module("nshtrainer.loggers").CSVLoggerConfig
         if name == "LoggerConfig":
             return importlib.import_module("nshtrainer.loggers").LoggerConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
@@ -53,6 +56,7 @@ else:
 
 # Submodule exports
 from . import _base as _base
+from . import actsave as actsave
 from . import csv as csv
 from . import tensorboard as tensorboard
 from . import wandb as wandb

nshtrainer/config/loggers/actsave/__init__.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from typing import TYPE_CHECKING
+
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.loggers.actsave import ActSaveLoggerConfig as ActSaveLoggerConfig
+    from nshtrainer.loggers.actsave import BaseLoggerConfig as BaseLoggerConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "ActSaveLoggerConfig":
+            return importlib.import_module(
+                "nshtrainer.loggers.actsave"
+            ).ActSaveLoggerConfig
+        if name == "BaseLoggerConfig":
+            return importlib.import_module(
+                "nshtrainer.loggers.actsave"
+            ).BaseLoggerConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
+
+# Submodule exports

nshtrainer/config/loggers/csv/__init__.py

@@ -16,10 +16,10 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "CSVLoggerConfig":
-            return importlib.import_module("nshtrainer.loggers.csv").CSVLoggerConfig
         if name == "BaseLoggerConfig":
             return importlib.import_module("nshtrainer.loggers.csv").BaseLoggerConfig
+        if name == "CSVLoggerConfig":
+            return importlib.import_module("nshtrainer.loggers.csv").CSVLoggerConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/loggers/wandb/__init__.py

@@ -23,6 +23,12 @@ else:
 
         if name in globals():
             return globals()[name]
+        if name == "BaseLoggerConfig":
+            return importlib.import_module("nshtrainer.loggers.wandb").BaseLoggerConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.loggers.wandb"
+            ).CallbackConfigBase
         if name == "WandbLoggerConfig":
             return importlib.import_module("nshtrainer.loggers.wandb").WandbLoggerConfig
         if name == "WandbUploadCodeCallbackConfig":
@@ -33,12 +39,6 @@ else:
             return importlib.import_module(
                 "nshtrainer.loggers.wandb"
             ).WandbWatchCallbackConfig
-        if name == "BaseLoggerConfig":
-            return importlib.import_module("nshtrainer.loggers.wandb").BaseLoggerConfig
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.loggers.wandb"
-            ).CallbackConfigBase
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py

@@ -23,14 +23,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "LinearWarmupCosineDecayLRSchedulerConfig":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler.linear_warmup_cosine"
-            ).LinearWarmupCosineDecayLRSchedulerConfig
         if name == "LRSchedulerConfigBase":
             return importlib.import_module(
                 "nshtrainer.lr_scheduler.linear_warmup_cosine"
             ).LRSchedulerConfigBase
+        if name == "LinearWarmupCosineDecayLRSchedulerConfig":
+            return importlib.import_module(
+                "nshtrainer.lr_scheduler.linear_warmup_cosine"
+            ).LinearWarmupCosineDecayLRSchedulerConfig
         if name == "DurationConfig":
             return importlib.import_module(
                 "nshtrainer.lr_scheduler.linear_warmup_cosine"

nshtrainer/config/nn/__init__.py

@@ -35,38 +35,38 @@ else:
             return globals()[name]
         if name == "BaseNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").BaseNonlinearityConfig
+        if name == "ELUNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").ELUNonlinearityConfig
+        if name == "GELUNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").GELUNonlinearityConfig
+        if name == "LeakyReLUNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").LeakyReLUNonlinearityConfig
         if name == "MLPConfig":
             return importlib.import_module("nshtrainer.nn").MLPConfig
+        if name == "MishNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").MishNonlinearityConfig
         if name == "PReLUConfig":
             return importlib.import_module("nshtrainer.nn").PReLUConfig
-        if name == "LeakyReLUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").LeakyReLUNonlinearityConfig
-        if name == "SwiGLUNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).SwiGLUNonlinearityConfig
-        if name == "SoftsignNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SoftsignNonlinearityConfig
+        if name == "ReLUNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").ReLUNonlinearityConfig
         if name == "SiLUNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").SiLUNonlinearityConfig
         if name == "SigmoidNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").SigmoidNonlinearityConfig
-        if name == "SoftplusNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SoftplusNonlinearityConfig
-        if name == "ELUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").ELUNonlinearityConfig
         if name == "SoftmaxNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").SoftmaxNonlinearityConfig
-        if name == "GELUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").GELUNonlinearityConfig
+        if name == "SoftplusNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").SoftplusNonlinearityConfig
+        if name == "SoftsignNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").SoftsignNonlinearityConfig
+        if name == "SwiGLUNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).SwiGLUNonlinearityConfig
         if name == "SwishNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").SwishNonlinearityConfig
-        if name == "MishNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").MishNonlinearityConfig
         if name == "TanhNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").TanhNonlinearityConfig
-        if name == "ReLUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").ReLUNonlinearityConfig
         if name == "NonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").NonlinearityConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

nshtrainer/config/nn/nonlinearity/__init__.py

@@ -58,20 +58,32 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "PReLUConfig":
-            return importlib.import_module("nshtrainer.nn.nonlinearity").PReLUConfig
+        if name == "BaseNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).BaseNonlinearityConfig
+        if name == "ELUNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).ELUNonlinearityConfig
+        if name == "GELUNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).GELUNonlinearityConfig
         if name == "LeakyReLUNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
             ).LeakyReLUNonlinearityConfig
-        if name == "SwiGLUNonlinearityConfig":
+        if name == "MishNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).SwiGLUNonlinearityConfig
-        if name == "SoftsignNonlinearityConfig":
+            ).MishNonlinearityConfig
+        if name == "PReLUConfig":
+            return importlib.import_module("nshtrainer.nn.nonlinearity").PReLUConfig
+        if name == "ReLUNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).SoftsignNonlinearityConfig
+            ).ReLUNonlinearityConfig
         if name == "SiLUNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
@@ -80,42 +92,30 @@ else:
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
             ).SigmoidNonlinearityConfig
-        if name == "SoftplusNonlinearityConfig":
+        if name == "SoftmaxNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).SoftplusNonlinearityConfig
-        if name == "ELUNonlinearityConfig":
+            ).SoftmaxNonlinearityConfig
+        if name == "SoftplusNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).ELUNonlinearityConfig
-        if name == "SoftmaxNonlinearityConfig":
+            ).SoftplusNonlinearityConfig
+        if name == "SoftsignNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).SoftmaxNonlinearityConfig
-        if name == "GELUNonlinearityConfig":
+            ).SoftsignNonlinearityConfig
+        if name == "SwiGLUNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).GELUNonlinearityConfig
+            ).SwiGLUNonlinearityConfig
         if name == "SwishNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
             ).SwishNonlinearityConfig
-        if name == "MishNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).MishNonlinearityConfig
         if name == "TanhNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
             ).TanhNonlinearityConfig
-        if name == "ReLUNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).ReLUNonlinearityConfig
-        if name == "BaseNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).BaseNonlinearityConfig
         if name == "NonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"

nshtrainer/config/optimizer/__init__.py

@@ -17,10 +17,10 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "OptimizerConfigBase":
-            return importlib.import_module("nshtrainer.optimizer").OptimizerConfigBase
         if name == "AdamWConfig":
             return importlib.import_module("nshtrainer.optimizer").AdamWConfig
+        if name == "OptimizerConfigBase":
+            return importlib.import_module("nshtrainer.optimizer").OptimizerConfigBase
         if name == "OptimizerConfig":
             return importlib.import_module("nshtrainer.optimizer").OptimizerConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

nshtrainer/config/profiler/__init__.py

@@ -19,12 +19,12 @@ else:
 
         if name in globals():
             return globals()[name]
+        if name == "AdvancedProfilerConfig":
+            return importlib.import_module("nshtrainer.profiler").AdvancedProfilerConfig
         if name == "BaseProfilerConfig":
             return importlib.import_module("nshtrainer.profiler").BaseProfilerConfig
         if name == "PyTorchProfilerConfig":
             return importlib.import_module("nshtrainer.profiler").PyTorchProfilerConfig
-        if name == "AdvancedProfilerConfig":
-            return importlib.import_module("nshtrainer.profiler").AdvancedProfilerConfig
         if name == "SimpleProfilerConfig":
             return importlib.import_module("nshtrainer.profiler").SimpleProfilerConfig
         if name == "ProfilerConfig":

nshtrainer/config/profiler/pytorch/__init__.py

@@ -18,14 +18,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "PyTorchProfilerConfig":
-            return importlib.import_module(
-                "nshtrainer.profiler.pytorch"
-            ).PyTorchProfilerConfig
         if name == "BaseProfilerConfig":
             return importlib.import_module(
                 "nshtrainer.profiler.pytorch"
             ).BaseProfilerConfig
+        if name == "PyTorchProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler.pytorch"
+            ).PyTorchProfilerConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/profiler/simple/__init__.py

@@ -16,14 +16,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "SimpleProfilerConfig":
-            return importlib.import_module(
-                "nshtrainer.profiler.simple"
-            ).SimpleProfilerConfig
         if name == "BaseProfilerConfig":
             return importlib.import_module(
                 "nshtrainer.profiler.simple"
             ).BaseProfilerConfig
+        if name == "SimpleProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler.simple"
+            ).SimpleProfilerConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/trainer/__init__.py

@@ -0,0 +1,180 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from typing import TYPE_CHECKING
+
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.trainer import TrainerConfig as TrainerConfig
+    from nshtrainer.trainer._config import ActSaveLoggerConfig as ActSaveLoggerConfig
+    from nshtrainer.trainer._config import (
+        BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
+    )
+    from nshtrainer.trainer._config import CallbackConfig as CallbackConfig
+    from nshtrainer.trainer._config import CallbackConfigBase as CallbackConfigBase
+    from nshtrainer.trainer._config import (
+        CheckpointCallbackConfig as CheckpointCallbackConfig,
+    )
+    from nshtrainer.trainer._config import (
+        CheckpointLoadingConfig as CheckpointLoadingConfig,
+    )
+    from nshtrainer.trainer._config import (
+        CheckpointSavingConfig as CheckpointSavingConfig,
+    )
+    from nshtrainer.trainer._config import CSVLoggerConfig as CSVLoggerConfig
+    from nshtrainer.trainer._config import (
+        DebugFlagCallbackConfig as DebugFlagCallbackConfig,
+    )
+    from nshtrainer.trainer._config import DirectoryConfig as DirectoryConfig
+    from nshtrainer.trainer._config import (
+        EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
+    )
+    from nshtrainer.trainer._config import EnvironmentConfig as EnvironmentConfig
+    from nshtrainer.trainer._config import (
+        GradientClippingConfig as GradientClippingConfig,
+    )
+    from nshtrainer.trainer._config import HuggingFaceHubConfig as HuggingFaceHubConfig
+    from nshtrainer.trainer._config import (
+        LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+    )
+    from nshtrainer.trainer._config import (
+        LogEpochCallbackConfig as LogEpochCallbackConfig,
+    )
+    from nshtrainer.trainer._config import LoggerConfig as LoggerConfig
+    from nshtrainer.trainer._config import LoggingConfig as LoggingConfig
+    from nshtrainer.trainer._config import MetricConfig as MetricConfig
+    from nshtrainer.trainer._config import (
+        OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+    )
+    from nshtrainer.trainer._config import OptimizationConfig as OptimizationConfig
+    from nshtrainer.trainer._config import ProfilerConfig as ProfilerConfig
+    from nshtrainer.trainer._config import (
+        ReproducibilityConfig as ReproducibilityConfig,
+    )
+    from nshtrainer.trainer._config import (
+        RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
+    )
+    from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
+    from nshtrainer.trainer._config import (
+        SharedParametersCallbackConfig as SharedParametersCallbackConfig,
+    )
+    from nshtrainer.trainer._config import (
+        TensorboardLoggerConfig as TensorboardLoggerConfig,
+    )
+    from nshtrainer.trainer._config import WandbLoggerConfig as WandbLoggerConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "ActSaveLoggerConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).ActSaveLoggerConfig
+        if name == "BestCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).BestCheckpointCallbackConfig
+        if name == "CSVLoggerConfig":
+            return importlib.import_module("nshtrainer.trainer._config").CSVLoggerConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).CallbackConfigBase
+        if name == "CheckpointLoadingConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).CheckpointLoadingConfig
+        if name == "CheckpointSavingConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).CheckpointSavingConfig
+        if name == "DebugFlagCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).DebugFlagCallbackConfig
+        if name == "DirectoryConfig":
+            return importlib.import_module("nshtrainer.trainer._config").DirectoryConfig
+        if name == "EarlyStoppingCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).EarlyStoppingCallbackConfig
+        if name == "EnvironmentConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).EnvironmentConfig
+        if name == "GradientClippingConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).GradientClippingConfig
+        if name == "HuggingFaceHubConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).HuggingFaceHubConfig
+        if name == "LastCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).LastCheckpointCallbackConfig
+        if name == "LogEpochCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).LogEpochCallbackConfig
+        if name == "LoggingConfig":
+            return importlib.import_module("nshtrainer.trainer._config").LoggingConfig
+        if name == "MetricConfig":
+            return importlib.import_module("nshtrainer.trainer._config").MetricConfig
+        if name == "OnExceptionCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).OnExceptionCheckpointCallbackConfig
+        if name == "OptimizationConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).OptimizationConfig
+        if name == "RLPSanityChecksCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).RLPSanityChecksCallbackConfig
+        if name == "ReproducibilityConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).ReproducibilityConfig
+        if name == "SanityCheckingConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).SanityCheckingConfig
+        if name == "SharedParametersCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).SharedParametersCallbackConfig
+        if name == "TensorboardLoggerConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).TensorboardLoggerConfig
+        if name == "TrainerConfig":
+            return importlib.import_module("nshtrainer.trainer").TrainerConfig
+        if name == "WandbLoggerConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).WandbLoggerConfig
+        if name == "CallbackConfig":
+            return importlib.import_module("nshtrainer.trainer._config").CallbackConfig
+        if name == "CheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).CheckpointCallbackConfig
+        if name == "LoggerConfig":
+            return importlib.import_module("nshtrainer.trainer._config").LoggerConfig
+        if name == "ProfilerConfig":
+            return importlib.import_module("nshtrainer.trainer._config").ProfilerConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
+
+
+# Submodule exports
+from . import _config as _config
+from . import checkpoint_connector as checkpoint_connector
+from . import trainer as trainer