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

nshtrainer/config.py

@@ -1,171 +0,0 @@
-from nshtrainer._checkpoint.loader import (
-    BestCheckpointStrategyConfig as BestCheckpointStrategyConfig,
-)
-from nshtrainer._checkpoint.loader import (
-    CheckpointLoadingConfig as CheckpointLoadingConfig,
-)
-from nshtrainer._checkpoint.loader import (
-    CheckpointLoadingStrategyConfig as CheckpointLoadingStrategyConfig,
-)
-from nshtrainer._checkpoint.loader import (
-    LastCheckpointStrategyConfig as LastCheckpointStrategyConfig,
-)
-from nshtrainer._checkpoint.loader import (
-    UserProvidedPathCheckpointStrategyConfig as UserProvidedPathCheckpointStrategyConfig,
-)
-from nshtrainer._checkpoint.metadata import CheckpointMetadata as CheckpointMetadata
-from nshtrainer._directory import DirectoryConfig as DirectoryConfig
-from nshtrainer._hf_hub import (
-    HuggingFaceHubAutoCreateConfig as HuggingFaceHubAutoCreateConfig,
-)
-from nshtrainer._hf_hub import HuggingFaceHubConfig as HuggingFaceHubConfig
-from nshtrainer.callbacks.actsave import ActSaveConfig as ActSaveConfig
-from nshtrainer.callbacks.base import CallbackConfigBase as CallbackConfigBase
-from nshtrainer.callbacks.checkpoint._base import (
-    BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig,
-)
-from nshtrainer.callbacks.checkpoint.best_checkpoint import (
-    BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
-)
-from nshtrainer.callbacks.checkpoint.last_checkpoint import (
-    LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
-)
-from nshtrainer.callbacks.checkpoint.on_exception_checkpoint import (
-    OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
-)
-from nshtrainer.callbacks.debug_flag import (
-    DebugFlagCallbackConfig as DebugFlagCallbackConfig,
-)
-from nshtrainer.callbacks.directory_setup import (
-    DirectorySetupConfig as DirectorySetupConfig,
-)
-from nshtrainer.callbacks.early_stopping import (
-    EarlyStoppingConfig as EarlyStoppingConfig,
-)
-from nshtrainer.callbacks.ema import EMAConfig as EMAConfig
-from nshtrainer.callbacks.finite_checks import FiniteChecksConfig as FiniteChecksConfig
-from nshtrainer.callbacks.gradient_skipping import (
-    GradientSkippingConfig as GradientSkippingConfig,
-)
-from nshtrainer.callbacks.norm_logging import NormLoggingConfig as NormLoggingConfig
-from nshtrainer.callbacks.print_table import (
-    PrintTableMetricsConfig as PrintTableMetricsConfig,
-)
-from nshtrainer.callbacks.rlp_sanity_checks import (
-    RLPSanityChecksConfig as RLPSanityChecksConfig,
-)
-from nshtrainer.callbacks.shared_parameters import (
-    SharedParametersConfig as SharedParametersConfig,
-)
-from nshtrainer.callbacks.throughput_monitor import (
-    ThroughputMonitorConfig as ThroughputMonitorConfig,
-)
-from nshtrainer.callbacks.timer import EpochTimerConfig as EpochTimerConfig
-from nshtrainer.callbacks.wandb_upload_code import (
-    WandbUploadCodeConfig as WandbUploadCodeConfig,
-)
-from nshtrainer.callbacks.wandb_watch import WandbWatchConfig as WandbWatchConfig
-from nshtrainer.loggers._base import BaseLoggerConfig as BaseLoggerConfig
-from nshtrainer.loggers.csv import CSVLoggerConfig as CSVLoggerConfig
-from nshtrainer.loggers.tensorboard import (
-    TensorboardLoggerConfig as TensorboardLoggerConfig,
-)
-from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
-from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
-from nshtrainer.lr_scheduler._base import LRSchedulerConfigBase as LRSchedulerConfigBase
-from nshtrainer.lr_scheduler.linear_warmup_cosine import (
-    DurationConfig as DurationConfig,
-)
-from nshtrainer.lr_scheduler.linear_warmup_cosine import (
-    LinearWarmupCosineDecayLRSchedulerConfig as LinearWarmupCosineDecayLRSchedulerConfig,
-)
-from nshtrainer.lr_scheduler.reduce_lr_on_plateau import (
-    ReduceLROnPlateauConfig as ReduceLROnPlateauConfig,
-)
-from nshtrainer.metrics._config import MetricConfig as MetricConfig
-from nshtrainer.model.config import BaseConfig as BaseConfig
-from nshtrainer.nn.mlp import MLPConfig as MLPConfig
-from nshtrainer.nn.nonlinearity import BaseNonlinearityConfig as BaseNonlinearityConfig
-from nshtrainer.nn.nonlinearity import ELUNonlinearityConfig as ELUNonlinearityConfig
-from nshtrainer.nn.nonlinearity import GELUNonlinearityConfig as GELUNonlinearityConfig
-from nshtrainer.nn.nonlinearity import (
-    LeakyReLUNonlinearityConfig as LeakyReLUNonlinearityConfig,
-)
-from nshtrainer.nn.nonlinearity import MishNonlinearityConfig as MishNonlinearityConfig
-from nshtrainer.nn.nonlinearity import NonlinearityConfig as NonlinearityConfig
-from nshtrainer.nn.nonlinearity import PReLUConfig as PReLUConfig
-from nshtrainer.nn.nonlinearity import ReLUNonlinearityConfig as ReLUNonlinearityConfig
-from nshtrainer.nn.nonlinearity import (
-    SigmoidNonlinearityConfig as SigmoidNonlinearityConfig,
-)
-from nshtrainer.nn.nonlinearity import SiLUNonlinearityConfig as SiLUNonlinearityConfig
-from nshtrainer.nn.nonlinearity import (
-    SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig,
-)
-from nshtrainer.nn.nonlinearity import (
-    SoftplusNonlinearityConfig as SoftplusNonlinearityConfig,
-)
-from nshtrainer.nn.nonlinearity import (
-    SoftsignNonlinearityConfig as SoftsignNonlinearityConfig,
-)
-from nshtrainer.nn.nonlinearity import (
-    SwiGLUNonlinearityConfig as SwiGLUNonlinearityConfig,
-)
-from nshtrainer.nn.nonlinearity import (
-    SwishNonlinearityConfig as SwishNonlinearityConfig,
-)
-from nshtrainer.nn.nonlinearity import TanhNonlinearityConfig as TanhNonlinearityConfig
-from nshtrainer.optimizer import AdamWConfig as AdamWConfig
-from nshtrainer.optimizer import OptimizerConfig as OptimizerConfig
-from nshtrainer.optimizer import OptimizerConfigBase as OptimizerConfigBase
-from nshtrainer.profiler._base import BaseProfilerConfig as BaseProfilerConfig
-from nshtrainer.profiler.advanced import (
-    AdvancedProfilerConfig as AdvancedProfilerConfig,
-)
-from nshtrainer.profiler.pytorch import PyTorchProfilerConfig as PyTorchProfilerConfig
-from nshtrainer.profiler.simple import SimpleProfilerConfig as SimpleProfilerConfig
-from nshtrainer.trainer._config import CallbackConfig as CallbackConfig
-from nshtrainer.trainer._config import (
-    CheckpointCallbackConfig as CheckpointCallbackConfig,
-)
-from nshtrainer.trainer._config import CheckpointSavingConfig as CheckpointSavingConfig
-from nshtrainer.trainer._config import GradientClippingConfig as GradientClippingConfig
-from nshtrainer.trainer._config import LoggerConfig as LoggerConfig
-from nshtrainer.trainer._config import LoggingConfig as LoggingConfig
-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 SanityCheckingConfig as SanityCheckingConfig
-from nshtrainer.trainer._config import TrainerConfig as TrainerConfig
-from nshtrainer.util._environment_info import (
-    EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
-)
-from nshtrainer.util._environment_info import EnvironmentConfig as EnvironmentConfig
-from nshtrainer.util._environment_info import (
-    EnvironmentCUDAConfig as EnvironmentCUDAConfig,
-)
-from nshtrainer.util._environment_info import (
-    EnvironmentGPUConfig as EnvironmentGPUConfig,
-)
-from nshtrainer.util._environment_info import (
-    EnvironmentHardwareConfig as EnvironmentHardwareConfig,
-)
-from nshtrainer.util._environment_info import (
-    EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
-)
-from nshtrainer.util._environment_info import (
-    EnvironmentLSFInformationConfig as EnvironmentLSFInformationConfig,
-)
-from nshtrainer.util._environment_info import (
-    EnvironmentPackageConfig as EnvironmentPackageConfig,
-)
-from nshtrainer.util._environment_info import (
-    EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig,
-)
-from nshtrainer.util._environment_info import (
-    EnvironmentSnapshotConfig as EnvironmentSnapshotConfig,
-)
-from nshtrainer.util._environment_info import GitRepositoryConfig as GitRepositoryConfig
-from nshtrainer.util.config.dtype import DTypeConfig as DTypeConfig
-from nshtrainer.util.config.duration import EpochsConfig as EpochsConfig
-from nshtrainer.util.config.duration import StepsConfig as StepsConfig