nshtrainer 1.0.0b14__tar.gz → 1.0.0b16__tar.gz

{nshtrainer-1.0.0b14 → nshtrainer-1.0.0b16}/PKG-INFO

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

{nshtrainer-1.0.0b14 → nshtrainer-1.0.0b16}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshtrainer"
-version = "1.0.0-beta14"
+version = "1.0.0-beta16"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"

nshtrainer-1.0.0b16/src/nshtrainer/configs/__init__.py

@@ -0,0 +1,151 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer import MetricConfig as MetricConfig
+from nshtrainer import TrainerConfig as TrainerConfig
+from nshtrainer._checkpoint.metadata import CheckpointMetadata as CheckpointMetadata
+from nshtrainer._directory import DirectoryConfig as DirectoryConfig
+from nshtrainer._hf_hub import CallbackConfigBase as CallbackConfigBase
+from nshtrainer._hf_hub import (
+    HuggingFaceHubAutoCreateConfig as HuggingFaceHubAutoCreateConfig,
+)
+from nshtrainer._hf_hub import HuggingFaceHubConfig as HuggingFaceHubConfig
+from nshtrainer.callbacks import (
+    BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks import CallbackConfig as CallbackConfig
+from nshtrainer.callbacks import DebugFlagCallbackConfig as DebugFlagCallbackConfig
+from nshtrainer.callbacks import (
+    DirectorySetupCallbackConfig as DirectorySetupCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
+)
+from nshtrainer.callbacks import EMACallbackConfig as EMACallbackConfig
+from nshtrainer.callbacks import EpochTimerCallbackConfig as EpochTimerCallbackConfig
+from nshtrainer.callbacks import (
+    FiniteChecksCallbackConfig as FiniteChecksCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    GradientSkippingCallbackConfig as GradientSkippingCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks import LogEpochCallbackConfig as LogEpochCallbackConfig
+from nshtrainer.callbacks import NormLoggingCallbackConfig as NormLoggingCallbackConfig
+from nshtrainer.callbacks import (
+    OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    PrintTableMetricsCallbackConfig as PrintTableMetricsCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    SharedParametersCallbackConfig as SharedParametersCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
+)
+from nshtrainer.callbacks import WandbWatchCallbackConfig as WandbWatchCallbackConfig
+from nshtrainer.callbacks.actsave import ActSaveConfig as ActSaveConfig
+from nshtrainer.callbacks.checkpoint._base import (
+    BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig,
+)
+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
+from nshtrainer.loggers import TensorboardLoggerConfig as TensorboardLoggerConfig
+from nshtrainer.loggers import WandbLoggerConfig as WandbLoggerConfig
+from nshtrainer.lr_scheduler import (
+    LinearWarmupCosineDecayLRSchedulerConfig as LinearWarmupCosineDecayLRSchedulerConfig,
+)
+from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
+from nshtrainer.lr_scheduler import LRSchedulerConfigBase as LRSchedulerConfigBase
+from nshtrainer.lr_scheduler import ReduceLROnPlateauConfig as ReduceLROnPlateauConfig
+from nshtrainer.nn import BaseNonlinearityConfig as BaseNonlinearityConfig
+from nshtrainer.nn import ELUNonlinearityConfig as ELUNonlinearityConfig
+from nshtrainer.nn import GELUNonlinearityConfig as GELUNonlinearityConfig
+from nshtrainer.nn import LeakyReLUNonlinearityConfig as LeakyReLUNonlinearityConfig
+from nshtrainer.nn import MishNonlinearityConfig as MishNonlinearityConfig
+from nshtrainer.nn import MLPConfig as MLPConfig
+from nshtrainer.nn import NonlinearityConfig as NonlinearityConfig
+from nshtrainer.nn import PReLUConfig as PReLUConfig
+from nshtrainer.nn import ReLUNonlinearityConfig as ReLUNonlinearityConfig
+from nshtrainer.nn import SigmoidNonlinearityConfig as SigmoidNonlinearityConfig
+from nshtrainer.nn import SiLUNonlinearityConfig as SiLUNonlinearityConfig
+from nshtrainer.nn import SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig
+from nshtrainer.nn import SoftplusNonlinearityConfig as SoftplusNonlinearityConfig
+from nshtrainer.nn import SoftsignNonlinearityConfig as SoftsignNonlinearityConfig
+from nshtrainer.nn import SwishNonlinearityConfig as SwishNonlinearityConfig
+from nshtrainer.nn import TanhNonlinearityConfig as TanhNonlinearityConfig
+from nshtrainer.nn.nonlinearity import (
+    SwiGLUNonlinearityConfig as SwiGLUNonlinearityConfig,
+)
+from nshtrainer.optimizer import AdamWConfig as AdamWConfig
+from nshtrainer.optimizer import OptimizerConfig as OptimizerConfig
+from nshtrainer.optimizer import OptimizerConfigBase as OptimizerConfigBase
+from nshtrainer.profiler import AdvancedProfilerConfig as AdvancedProfilerConfig
+from nshtrainer.profiler import BaseProfilerConfig as BaseProfilerConfig
+from nshtrainer.profiler import ProfilerConfig as ProfilerConfig
+from nshtrainer.profiler import PyTorchProfilerConfig as PyTorchProfilerConfig
+from nshtrainer.profiler import SimpleProfilerConfig as SimpleProfilerConfig
+from nshtrainer.trainer._config import (
+    CheckpointCallbackConfig as CheckpointCallbackConfig,
+)
+from nshtrainer.trainer._config import CheckpointSavingConfig as CheckpointSavingConfig
+from nshtrainer.trainer._config import EnvironmentConfig as EnvironmentConfig
+from nshtrainer.trainer._config import GradientClippingConfig as GradientClippingConfig
+from nshtrainer.trainer._config import (
+    LearningRateMonitorConfig as LearningRateMonitorConfig,
+)
+from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
+from nshtrainer.util._environment_info import (
+    EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
+)
+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 import DTypeConfig as DTypeConfig
+from nshtrainer.util.config import DurationConfig as DurationConfig
+from nshtrainer.util.config import EpochsConfig as EpochsConfig
+from nshtrainer.util.config import StepsConfig as StepsConfig
+
+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 nn as nn
+from . import optimizer as optimizer
+from . import profiler as profiler
+from . import trainer as trainer
+from . import util as util

nshtrainer-1.0.0b16/src/nshtrainer/configs/_checkpoint/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer._checkpoint.metadata import CheckpointMetadata as CheckpointMetadata
+from nshtrainer._checkpoint.metadata import EnvironmentConfig as EnvironmentConfig
+
+from . import metadata as metadata

nshtrainer-1.0.0b16/src/nshtrainer/configs/_checkpoint/metadata/__init__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer._checkpoint.metadata import CheckpointMetadata as CheckpointMetadata
+from nshtrainer._checkpoint.metadata import EnvironmentConfig as EnvironmentConfig

nshtrainer-1.0.0b16/src/nshtrainer/configs/_directory/__init__.py

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer._directory import DirectoryConfig as DirectoryConfig
+from nshtrainer._directory import (
+    DirectorySetupCallbackConfig as DirectorySetupCallbackConfig,
+)
+from nshtrainer._directory import LoggerConfig as LoggerConfig

nshtrainer-1.0.0b16/src/nshtrainer/configs/_hf_hub/__init__.py

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer._hf_hub import CallbackConfigBase as CallbackConfigBase
+from nshtrainer._hf_hub import (
+    HuggingFaceHubAutoCreateConfig as HuggingFaceHubAutoCreateConfig,
+)
+from nshtrainer._hf_hub import HuggingFaceHubConfig as HuggingFaceHubConfig

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/__init__.py

@@ -0,0 +1,75 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks import (
+    BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks import CallbackConfig as CallbackConfig
+from nshtrainer.callbacks import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks import DebugFlagCallbackConfig as DebugFlagCallbackConfig
+from nshtrainer.callbacks import (
+    DirectorySetupCallbackConfig as DirectorySetupCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
+)
+from nshtrainer.callbacks import EMACallbackConfig as EMACallbackConfig
+from nshtrainer.callbacks import EpochTimerCallbackConfig as EpochTimerCallbackConfig
+from nshtrainer.callbacks import (
+    FiniteChecksCallbackConfig as FiniteChecksCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    GradientSkippingCallbackConfig as GradientSkippingCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks import LogEpochCallbackConfig as LogEpochCallbackConfig
+from nshtrainer.callbacks import NormLoggingCallbackConfig as NormLoggingCallbackConfig
+from nshtrainer.callbacks import (
+    OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    PrintTableMetricsCallbackConfig as PrintTableMetricsCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    SharedParametersCallbackConfig as SharedParametersCallbackConfig,
+)
+from nshtrainer.callbacks import (
+    WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
+)
+from nshtrainer.callbacks import WandbWatchCallbackConfig as WandbWatchCallbackConfig
+from nshtrainer.callbacks.actsave import ActSaveConfig as ActSaveConfig
+from nshtrainer.callbacks.checkpoint._base import (
+    BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks.checkpoint._base import (
+    CheckpointMetadata as CheckpointMetadata,
+)
+from nshtrainer.callbacks.early_stopping import MetricConfig as MetricConfig
+from nshtrainer.callbacks.lr_monitor import (
+    LearningRateMonitorConfig as LearningRateMonitorConfig,
+)
+
+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 log_epoch as log_epoch
+from . import lr_monitor as lr_monitor
+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 timer as timer
+from . import wandb_upload_code as wandb_upload_code
+from . import wandb_watch as wandb_watch

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/actsave/__init__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.actsave import ActSaveConfig as ActSaveConfig
+from nshtrainer.callbacks.actsave import CallbackConfigBase as CallbackConfigBase

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/base/__init__.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.base import CallbackConfigBase as CallbackConfigBase

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/checkpoint/__init__.py

@@ -0,0 +1,28 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.checkpoint import (
+    BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks.checkpoint import (
+    LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks.checkpoint import (
+    OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks.checkpoint._base import (
+    BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks.checkpoint._base import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.checkpoint._base import (
+    CheckpointMetadata as CheckpointMetadata,
+)
+from nshtrainer.callbacks.checkpoint.best_checkpoint import MetricConfig as MetricConfig
+
+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-1.0.0b16/src/nshtrainer/configs/callbacks/checkpoint/_base/__init__.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.checkpoint._base import (
+    BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks.checkpoint._base import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.checkpoint._base import (
+    CheckpointMetadata as CheckpointMetadata,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/checkpoint/best_checkpoint/__init__.py

@@ -0,0 +1,14 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.checkpoint.best_checkpoint import (
+    BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig,
+)
+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 MetricConfig as MetricConfig

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/checkpoint/last_checkpoint/__init__.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.checkpoint.last_checkpoint import (
+    BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig,
+)
+from nshtrainer.callbacks.checkpoint.last_checkpoint import (
+    CheckpointMetadata as CheckpointMetadata,
+)
+from nshtrainer.callbacks.checkpoint.last_checkpoint import (
+    LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/checkpoint/on_exception_checkpoint/__init__.py

@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.checkpoint.on_exception_checkpoint import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.checkpoint.on_exception_checkpoint import (
+    OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/debug_flag/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.debug_flag import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.debug_flag import (
+    DebugFlagCallbackConfig as DebugFlagCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/directory_setup/__init__.py

@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.directory_setup import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.directory_setup import (
+    DirectorySetupCallbackConfig as DirectorySetupCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/early_stopping/__init__.py

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.early_stopping import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.early_stopping import (
+    EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
+)
+from nshtrainer.callbacks.early_stopping import MetricConfig as MetricConfig

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/ema/__init__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.ema import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.ema import EMACallbackConfig as EMACallbackConfig

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/finite_checks/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.finite_checks import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.finite_checks import (
+    FiniteChecksCallbackConfig as FiniteChecksCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/gradient_skipping/__init__.py

@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.gradient_skipping import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.gradient_skipping import (
+    GradientSkippingCallbackConfig as GradientSkippingCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/log_epoch/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.log_epoch import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.log_epoch import (
+    LogEpochCallbackConfig as LogEpochCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/lr_monitor/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.lr_monitor import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.lr_monitor import (
+    LearningRateMonitorConfig as LearningRateMonitorConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/norm_logging/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.norm_logging import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.norm_logging import (
+    NormLoggingCallbackConfig as NormLoggingCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/print_table/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.print_table import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.print_table import (
+    PrintTableMetricsCallbackConfig as PrintTableMetricsCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/rlp_sanity_checks/__init__.py

@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.rlp_sanity_checks import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.rlp_sanity_checks import (
+    RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/shared_parameters/__init__.py

@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.shared_parameters import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.shared_parameters import (
+    SharedParametersCallbackConfig as SharedParametersCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/timer/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.timer import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.timer import (
+    EpochTimerCallbackConfig as EpochTimerCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/wandb_upload_code/__init__.py

@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.wandb_upload_code import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.wandb_upload_code import (
+    WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/callbacks/wandb_watch/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.callbacks.wandb_watch import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.wandb_watch import (
+    WandbWatchCallbackConfig as WandbWatchCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/loggers/__init__.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+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
+from nshtrainer.loggers import TensorboardLoggerConfig as TensorboardLoggerConfig
+from nshtrainer.loggers import WandbLoggerConfig as WandbLoggerConfig
+from nshtrainer.loggers.wandb import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.loggers.wandb import (
+    WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
+)
+from nshtrainer.loggers.wandb import (
+    WandbWatchCallbackConfig as WandbWatchCallbackConfig,
+)
+
+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-1.0.0b16/src/nshtrainer/configs/loggers/_base/__init__.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.loggers._base import BaseLoggerConfig as BaseLoggerConfig

nshtrainer-1.0.0b16/src/nshtrainer/configs/loggers/actsave/__init__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.loggers.actsave import ActSaveLoggerConfig as ActSaveLoggerConfig
+from nshtrainer.loggers.actsave import BaseLoggerConfig as BaseLoggerConfig

nshtrainer-1.0.0b16/src/nshtrainer/configs/loggers/csv/__init__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.loggers.csv import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.loggers.csv import CSVLoggerConfig as CSVLoggerConfig

nshtrainer-1.0.0b16/src/nshtrainer/configs/loggers/tensorboard/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.loggers.tensorboard import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.loggers.tensorboard import (
+    TensorboardLoggerConfig as TensorboardLoggerConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/loggers/wandb/__init__.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.loggers.wandb import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.loggers.wandb import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
+from nshtrainer.loggers.wandb import (
+    WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
+)
+from nshtrainer.loggers.wandb import (
+    WandbWatchCallbackConfig as WandbWatchCallbackConfig,
+)

nshtrainer-1.0.0b16/src/nshtrainer/configs/lr_scheduler/__init__.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.lr_scheduler import (
+    LinearWarmupCosineDecayLRSchedulerConfig as LinearWarmupCosineDecayLRSchedulerConfig,
+)
+from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
+from nshtrainer.lr_scheduler import LRSchedulerConfigBase as LRSchedulerConfigBase
+from nshtrainer.lr_scheduler import ReduceLROnPlateauConfig as ReduceLROnPlateauConfig
+from nshtrainer.lr_scheduler.linear_warmup_cosine import (
+    DurationConfig as DurationConfig,
+)
+from nshtrainer.lr_scheduler.reduce_lr_on_plateau import MetricConfig as MetricConfig
+
+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-1.0.0b16/src/nshtrainer/configs/lr_scheduler/_base/__init__.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+from nshtrainer.lr_scheduler._base import LRSchedulerConfigBase as LRSchedulerConfigBase

nshtrainer-1.0.0b16/src/nshtrainer/configs/lr_scheduler/linear_warmup_cosine/__init__.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+__codegen__ = True
+
+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.linear_warmup_cosine import (
+    LRSchedulerConfigBase as LRSchedulerConfigBase,
+)