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

nshtrainer/config/__init__.py

@@ -0,0 +1,114 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer import MetricConfig as MetricConfig
+from nshtrainer import BaseConfig as BaseConfig
+from nshtrainer._hf_hub import HuggingFaceHubAutoCreateConfig as HuggingFaceHubAutoCreateConfig
+from nshtrainer._hf_hub import CallbackConfigBase as CallbackConfigBase
+from nshtrainer._hf_hub import HuggingFaceHubConfig as HuggingFaceHubConfig
+from nshtrainer.optimizer import OptimizerConfigBase as OptimizerConfigBase
+from nshtrainer.optimizer import AdamWConfig as AdamWConfig
+from nshtrainer.model import DirectoryConfig as DirectoryConfig
+from nshtrainer.callbacks import DirectorySetupConfig as DirectorySetupConfig
+from nshtrainer.model import TrainerConfig as TrainerConfig
+from nshtrainer.model.base import EnvironmentConfig as EnvironmentConfig
+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
+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.loggers import TensorboardLoggerConfig as TensorboardLoggerConfig
+from nshtrainer.loggers import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.loggers import WandbLoggerConfig as WandbLoggerConfig
+from nshtrainer.callbacks import WandbUploadCodeConfig as WandbUploadCodeConfig
+from nshtrainer.callbacks import WandbWatchConfig as WandbWatchConfig
+from nshtrainer.loggers import CSVLoggerConfig as CSVLoggerConfig
+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 EnvironmentLSFInformationConfig as EnvironmentLSFInformationConfig
+from nshtrainer.util.config import EpochsConfig as EpochsConfig
+from nshtrainer.util.config import StepsConfig as StepsConfig
+from nshtrainer.util.config import DTypeConfig as DTypeConfig
+from nshtrainer.trainer._config import CheckpointLoadingConfig as CheckpointLoadingConfig
+from nshtrainer.callbacks import BestCheckpointCallbackConfig as BestCheckpointCallbackConfig
+from nshtrainer.trainer._config import CheckpointSavingConfig as CheckpointSavingConfig
+from nshtrainer.callbacks 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 LoggingConfig as LoggingConfig
+from nshtrainer.callbacks import LastCheckpointCallbackConfig as LastCheckpointCallbackConfig
+from nshtrainer.callbacks import OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig
+from nshtrainer.callbacks import RLPSanityChecksConfig as RLPSanityChecksConfig
+from nshtrainer.callbacks import EarlyStoppingConfig as EarlyStoppingConfig
+from nshtrainer.trainer._config import OptimizationConfig as OptimizationConfig
+from nshtrainer.callbacks import DebugFlagCallbackConfig as DebugFlagCallbackConfig
+from nshtrainer.trainer._config import GradientClippingConfig as GradientClippingConfig
+from nshtrainer._checkpoint.loader import CheckpointMetadata as CheckpointMetadata
+from nshtrainer._checkpoint.loader import UserProvidedPathCheckpointStrategyConfig as UserProvidedPathCheckpointStrategyConfig
+from nshtrainer._checkpoint.loader import BestCheckpointStrategyConfig as BestCheckpointStrategyConfig
+from nshtrainer._checkpoint.loader import LastCheckpointStrategyConfig as LastCheckpointStrategyConfig
+from nshtrainer.callbacks import PrintTableMetricsConfig as PrintTableMetricsConfig
+from nshtrainer.callbacks import ThroughputMonitorConfig as ThroughputMonitorConfig
+from nshtrainer.callbacks import GradientSkippingConfig as GradientSkippingConfig
+from nshtrainer.callbacks import EMAConfig as EMAConfig
+from nshtrainer.callbacks.actsave import ActSaveConfig as ActSaveConfig
+from nshtrainer.callbacks import FiniteChecksConfig as FiniteChecksConfig
+from nshtrainer.callbacks import NormLoggingConfig as NormLoggingConfig
+from nshtrainer.callbacks import EpochTimerConfig as EpochTimerConfig
+from nshtrainer.callbacks.checkpoint._base import BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig
+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.optimizer import OptimizerConfig as OptimizerConfig
+from nshtrainer.loggers import LoggerConfig as LoggerConfig
+from nshtrainer.nn import NonlinearityConfig as NonlinearityConfig
+from nshtrainer.util.config import DurationConfig as DurationConfig
+from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
+from nshtrainer.callbacks import CallbackConfig as CallbackConfig
+from nshtrainer.trainer._config import CheckpointCallbackConfig as CheckpointCallbackConfig
+from nshtrainer.profiler import ProfilerConfig as ProfilerConfig
+from nshtrainer._checkpoint.loader import CheckpointLoadingStrategyConfig as CheckpointLoadingStrategyConfig
+
+# Submodule exports
+from . import _checkpoint as _checkpoint
+from . import _directory as _directory
+from . import _hf_hub as _hf_hub
+from . import callbacks as callbacks
+from . import loggers as loggers
+from . import lr_scheduler as lr_scheduler
+from . import metrics as metrics
+from . import model as model
+from . import nn as nn
+from . import optimizer as optimizer
+from . import profiler as profiler
+from . import runner as runner
+from . import trainer as trainer
+from . import util as util

nshtrainer/config/_checkpoint/loader/__init__.py

@@ -0,0 +1,18 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer._checkpoint.loader import UserProvidedPathCheckpointStrategyConfig as UserProvidedPathCheckpointStrategyConfig
+from nshtrainer._checkpoint.loader import MetricConfig as MetricConfig
+from nshtrainer._checkpoint.loader import CheckpointLoadingConfig as CheckpointLoadingConfig
+from nshtrainer._checkpoint.loader import CheckpointMetadata as CheckpointMetadata
+from nshtrainer._checkpoint.loader import BestCheckpointStrategyConfig as BestCheckpointStrategyConfig
+from nshtrainer._checkpoint.loader import LastCheckpointStrategyConfig as LastCheckpointStrategyConfig
+
+# Type aliases
+from nshtrainer._checkpoint.loader import CheckpointLoadingStrategyConfig as CheckpointLoadingStrategyConfig
+
+# Submodule exports

nshtrainer/config/_checkpoint/metadata/__init__.py

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

nshtrainer/config/_directory/__init__.py

@@ -0,0 +1,14 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer._directory import DirectoryConfig as DirectoryConfig
+from nshtrainer._directory import DirectorySetupConfig as DirectorySetupConfig
+
+# Type aliases
+from nshtrainer._directory import LoggerConfig as LoggerConfig
+
+# Submodule exports

nshtrainer/config/_hf_hub/__init__.py

@@ -0,0 +1,14 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer._hf_hub import HuggingFaceHubAutoCreateConfig as HuggingFaceHubAutoCreateConfig
+from nshtrainer._hf_hub import CallbackConfigBase as CallbackConfigBase
+from nshtrainer._hf_hub import HuggingFaceHubConfig as HuggingFaceHubConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/callbacks/__init__.py

@@ -0,0 +1,51 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.callbacks import PrintTableMetricsConfig as PrintTableMetricsConfig
+from nshtrainer.callbacks import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks import DebugFlagCallbackConfig as DebugFlagCallbackConfig
+from nshtrainer.callbacks import ThroughputMonitorConfig as ThroughputMonitorConfig
+from nshtrainer.callbacks import GradientSkippingConfig as GradientSkippingConfig
+from nshtrainer.callbacks import RLPSanityChecksConfig as RLPSanityChecksConfig
+from nshtrainer.callbacks import WandbUploadCodeConfig as WandbUploadCodeConfig
+from nshtrainer.callbacks import EarlyStoppingConfig as EarlyStoppingConfig
+from nshtrainer.callbacks.early_stopping import MetricConfig as MetricConfig
+from nshtrainer.callbacks import WandbWatchConfig as WandbWatchConfig
+from nshtrainer.callbacks import EMAConfig as EMAConfig
+from nshtrainer.callbacks import DirectorySetupConfig as DirectorySetupConfig
+from nshtrainer.callbacks.actsave import ActSaveConfig as ActSaveConfig
+from nshtrainer.callbacks import FiniteChecksConfig as FiniteChecksConfig
+from nshtrainer.callbacks import NormLoggingConfig as NormLoggingConfig
+from nshtrainer.callbacks import BestCheckpointCallbackConfig as BestCheckpointCallbackConfig
+from nshtrainer.callbacks import OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig
+from nshtrainer.callbacks import SharedParametersConfig as SharedParametersConfig
+from nshtrainer.callbacks import EpochTimerConfig as EpochTimerConfig
+from nshtrainer.callbacks import LastCheckpointCallbackConfig as LastCheckpointCallbackConfig
+from nshtrainer.callbacks.checkpoint._base import CheckpointMetadata as CheckpointMetadata
+from nshtrainer.callbacks.checkpoint._base import BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig
+
+# Type aliases
+from nshtrainer.callbacks import CallbackConfig as CallbackConfig
+
+# Submodule exports
+from . import actsave as actsave
+from . import base as base
+from . import checkpoint as checkpoint
+from . import debug_flag as debug_flag
+from . import directory_setup as directory_setup
+from . import early_stopping as early_stopping
+from . import ema as ema
+from . import finite_checks as finite_checks
+from . import gradient_skipping as gradient_skipping
+from . import norm_logging as norm_logging
+from . import print_table as print_table
+from . import rlp_sanity_checks as rlp_sanity_checks
+from . import shared_parameters as shared_parameters
+from . import throughput_monitor as throughput_monitor
+from . import timer as timer
+from . import wandb_upload_code as wandb_upload_code
+from . import wandb_watch as wandb_watch

nshtrainer/config/callbacks/actsave/__init__.py

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

nshtrainer/config/callbacks/base/__init__.py

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

nshtrainer/config/callbacks/checkpoint/__init__.py

@@ -0,0 +1,22 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.callbacks.checkpoint import LastCheckpointCallbackConfig as LastCheckpointCallbackConfig
+from nshtrainer.callbacks.checkpoint._base import CheckpointMetadata as CheckpointMetadata
+from nshtrainer.callbacks.checkpoint._base import BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig
+from nshtrainer.callbacks.checkpoint._base import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.checkpoint import OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig
+from nshtrainer.callbacks.checkpoint import BestCheckpointCallbackConfig as BestCheckpointCallbackConfig
+from nshtrainer.callbacks.checkpoint.best_checkpoint import MetricConfig as MetricConfig
+
+# Type aliases
+
+# Submodule exports
+from . import _base as _base
+from . import best_checkpoint as best_checkpoint
+from . import last_checkpoint as last_checkpoint
+from . import on_exception_checkpoint as on_exception_checkpoint

nshtrainer/config/callbacks/checkpoint/_base/__init__.py

@@ -0,0 +1,14 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.callbacks.checkpoint._base import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.checkpoint._base import CheckpointMetadata as CheckpointMetadata
+from nshtrainer.callbacks.checkpoint._base import BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py

@@ -0,0 +1,15 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.callbacks.checkpoint.best_checkpoint import BestCheckpointCallbackConfig as BestCheckpointCallbackConfig
+from nshtrainer.callbacks.checkpoint.best_checkpoint import CheckpointMetadata as CheckpointMetadata
+from nshtrainer.callbacks.checkpoint.best_checkpoint import BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig
+from nshtrainer.callbacks.checkpoint.best_checkpoint import MetricConfig as MetricConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py

@@ -0,0 +1,14 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.callbacks.checkpoint.last_checkpoint import LastCheckpointCallbackConfig as LastCheckpointCallbackConfig
+from nshtrainer.callbacks.checkpoint.last_checkpoint import CheckpointMetadata as CheckpointMetadata
+from nshtrainer.callbacks.checkpoint.last_checkpoint import BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py

@@ -0,0 +1,13 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.callbacks.checkpoint.on_exception_checkpoint import OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig
+from nshtrainer.callbacks.checkpoint.on_exception_checkpoint import CallbackConfigBase as CallbackConfigBase
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/callbacks/debug_flag/__init__.py

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

nshtrainer/config/callbacks/directory_setup/__init__.py

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

nshtrainer/config/callbacks/early_stopping/__init__.py

@@ -0,0 +1,14 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.callbacks.early_stopping import EarlyStoppingConfig as EarlyStoppingConfig
+from nshtrainer.callbacks.early_stopping import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.early_stopping import MetricConfig as MetricConfig
+
+# Type aliases
+
+# Submodule exports

nshtrainer/config/callbacks/ema/__init__.py

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

nshtrainer/config/callbacks/finite_checks/__init__.py

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

nshtrainer/config/callbacks/gradient_skipping/__init__.py

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

nshtrainer/config/callbacks/norm_logging/__init__.py

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

nshtrainer/config/callbacks/print_table/__init__.py

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

nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py

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

nshtrainer/config/callbacks/shared_parameters/__init__.py

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

nshtrainer/config/callbacks/throughput_monitor/__init__.py

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

nshtrainer/config/callbacks/timer/__init__.py

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

nshtrainer/config/callbacks/wandb_upload_code/__init__.py

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

nshtrainer/config/callbacks/wandb_watch/__init__.py

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

nshtrainer/config/loggers/__init__.py

@@ -0,0 +1,23 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.loggers import TensorboardLoggerConfig as TensorboardLoggerConfig
+from nshtrainer.loggers import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.loggers import WandbLoggerConfig as WandbLoggerConfig
+from nshtrainer.loggers.wandb import WandbUploadCodeConfig as WandbUploadCodeConfig
+from nshtrainer.loggers.wandb import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.loggers.wandb import WandbWatchConfig as WandbWatchConfig
+from nshtrainer.loggers import CSVLoggerConfig as CSVLoggerConfig
+
+# Type aliases
+from nshtrainer.loggers import LoggerConfig as LoggerConfig
+
+# Submodule exports
+from . import _base as _base
+from . import csv as csv
+from . import tensorboard as tensorboard
+from . import wandb as wandb

nshtrainer/config/loggers/_base/__init__.py

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

nshtrainer/config/loggers/csv/__init__.py

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

nshtrainer/config/loggers/tensorboard/__init__.py

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

nshtrainer/config/loggers/wandb/__init__.py

@@ -0,0 +1,16 @@
+# fmt: off
+# ruff: noqa
+# type: ignore
+
+__codegen__ = True
+
+# Config classes
+from nshtrainer.loggers.wandb import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
+from nshtrainer.loggers.wandb import WandbUploadCodeConfig as WandbUploadCodeConfig
+from nshtrainer.loggers.wandb import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.loggers.wandb import WandbWatchConfig as WandbWatchConfig
+
+# Type aliases
+
+# Submodule exports

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