nshtrainer 1.0.0b33__py3-none-any.whl → 1.0.0b36__py3-none-any.whl

nshtrainer/configs/callbacks/wandb_upload_code/__init__.py

@@ -8,8 +8,12 @@ from nshtrainer.callbacks.wandb_upload_code import (
 from nshtrainer.callbacks.wandb_upload_code import (
     WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
 )
+from nshtrainer.callbacks.wandb_upload_code import (
+    callback_registry as callback_registry,
+)
 
 __all__ = [
     "CallbackConfigBase",
     "WandbUploadCodeCallbackConfig",
+    "callback_registry",
 ]

nshtrainer/configs/callbacks/wandb_watch/__init__.py

@@ -6,8 +6,10 @@ from nshtrainer.callbacks.wandb_watch import CallbackConfigBase as CallbackConfi
 from nshtrainer.callbacks.wandb_watch import (
     WandbWatchCallbackConfig as WandbWatchCallbackConfig,
 )
+from nshtrainer.callbacks.wandb_watch import callback_registry as callback_registry
 
 __all__ = [
     "CallbackConfigBase",
     "WandbWatchCallbackConfig",
+    "callback_registry",
 ]

nshtrainer/configs/trainer/__init__.py

@@ -4,6 +4,7 @@ __codegen__ = True
 
 from nshtrainer.trainer import TrainerConfig as TrainerConfig
 from nshtrainer.trainer import accelerator_registry as accelerator_registry
+from nshtrainer.trainer import callback_registry as callback_registry
 from nshtrainer.trainer import plugin_registry as plugin_registry
 from nshtrainer.trainer._config import AcceleratorConfig as AcceleratorConfig
 from nshtrainer.trainer._config import ActSaveLoggerConfig as ActSaveLoggerConfig
@@ -55,9 +56,6 @@ from nshtrainer.trainer._config import StrategyConfig as StrategyConfig
 from nshtrainer.trainer._config import (
     TensorboardLoggerConfig as TensorboardLoggerConfig,
 )
-from nshtrainer.trainer._config import (
-    TimeCheckpointCallbackConfig as TimeCheckpointCallbackConfig,
-)
 from nshtrainer.trainer._config import WandbLoggerConfig as WandbLoggerConfig
 from nshtrainer.trainer.accelerator import CPUAcceleratorConfig as CPUAcceleratorConfig
 from nshtrainer.trainer.accelerator import (
@@ -179,7 +177,6 @@ __all__ = [
     "StrategyConfig",
     "StrategyConfigBase",
     "TensorboardLoggerConfig",
-    "TimeCheckpointCallbackConfig",
     "TorchCheckpointIOPlugin",
     "TorchElasticEnvironmentPlugin",
     "TorchSyncBatchNormPlugin",
@@ -193,6 +190,7 @@ __all__ = [
     "_config",
     "accelerator",
     "accelerator_registry",
+    "callback_registry",
     "plugin",
     "plugin_registry",
     "strategy",

nshtrainer/configs/trainer/_config/__init__.py

@@ -53,13 +53,8 @@ from nshtrainer.trainer._config import StrategyConfig as StrategyConfig
 from nshtrainer.trainer._config import (
     TensorboardLoggerConfig as TensorboardLoggerConfig,
 )
-from nshtrainer.trainer._config import (
-    TimeCheckpointCallbackConfig as TimeCheckpointCallbackConfig,
-)
 from nshtrainer.trainer._config import TrainerConfig as TrainerConfig
 from nshtrainer.trainer._config import WandbLoggerConfig as WandbLoggerConfig
-from nshtrainer.trainer._config import accelerator_registry as accelerator_registry
-from nshtrainer.trainer._config import plugin_registry as plugin_registry
 
 __all__ = [
     "AcceleratorConfig",
@@ -91,9 +86,6 @@ __all__ = [
     "SharedParametersCallbackConfig",
     "StrategyConfig",
     "TensorboardLoggerConfig",
-    "TimeCheckpointCallbackConfig",
     "TrainerConfig",
     "WandbLoggerConfig",
-    "accelerator_registry",
-    "plugin_registry",
 ]

nshtrainer/trainer/__init__.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
+from ..callbacks import callback_registry as callback_registry
 from ._config import TrainerConfig as TrainerConfig
-from ._config import accelerator_registry as accelerator_registry
-from ._config import plugin_registry as plugin_registry
+from .accelerator import accelerator_registry as accelerator_registry
+from .plugin import plugin_registry as plugin_registry
 from .trainer import Trainer as Trainer

nshtrainer/trainer/_config.py

@@ -37,7 +37,6 @@ from ..callbacks import (
     OnExceptionCheckpointCallbackConfig,
 )
 from ..callbacks.base import CallbackConfigBase
-from ..callbacks.checkpoint.time_checkpoint import TimeCheckpointCallbackConfig
 from ..callbacks.debug_flag import DebugFlagCallbackConfig
 from ..callbacks.log_epoch import LogEpochCallbackConfig
 from ..callbacks.lr_monitor import LearningRateMonitorConfig
@@ -54,9 +53,9 @@ from ..loggers.actsave import ActSaveLoggerConfig
 from ..metrics._config import MetricConfig
 from ..profiler import ProfilerConfig
 from ..util._environment_info import EnvironmentConfig
-from .accelerator import AcceleratorConfig, AcceleratorLiteral, accelerator_registry
-from .plugin import PluginConfig, plugin_registry
-from .strategy import StrategyConfig
+from .accelerator import AcceleratorConfig, AcceleratorLiteral
+from .plugin import PluginConfig
+from .strategy import StrategyConfig, StrategyLiteral
 
 log = logging.getLogger(__name__)
 
@@ -70,46 +69,12 @@ class GradientClippingConfig(C.Config):
     """Norm type to use for gradient clipping."""
 
 
-StrategyLiteral = TypeAliasType(
-    "StrategyLiteral",
-    Literal[
-        "auto",
-        "ddp",
-        "ddp_find_unused_parameters_false",
-        "ddp_find_unused_parameters_true",
-        "ddp_spawn",
-        "ddp_spawn_find_unused_parameters_false",
-        "ddp_spawn_find_unused_parameters_true",
-        "ddp_fork",
-        "ddp_fork_find_unused_parameters_false",
-        "ddp_fork_find_unused_parameters_true",
-        "ddp_notebook",
-        "dp",
-        "deepspeed",
-        "deepspeed_stage_1",
-        "deepspeed_stage_1_offload",
-        "deepspeed_stage_2",
-        "deepspeed_stage_2_offload",
-        "deepspeed_stage_3",
-        "deepspeed_stage_3_offload",
-        "deepspeed_stage_3_offload_nvme",
-        "fsdp",
-        "fsdp_cpu_offload",
-        "single_xla",
-        "xla_fsdp",
-        "xla",
-        "single_tpu",
-    ],
-)
-
-
 CheckpointCallbackConfig = TypeAliasType(
     "CheckpointCallbackConfig",
     Annotated[
         BestCheckpointCallbackConfig
         | LastCheckpointCallbackConfig
-        | OnExceptionCheckpointCallbackConfig
-        | TimeCheckpointCallbackConfig,
+        | OnExceptionCheckpointCallbackConfig,
         C.Field(discriminator="name"),
     ],
 )
@@ -123,7 +88,6 @@ class CheckpointSavingConfig(CallbackConfigBase):
         BestCheckpointCallbackConfig(throw_on_no_metric=False),
         LastCheckpointCallbackConfig(),
         OnExceptionCheckpointCallbackConfig(),
-        TimeCheckpointCallbackConfig(interval=timedelta(hours=12)),
     ]
     """Checkpoint callback configurations."""
 
@@ -397,8 +361,6 @@ class SanityCheckingConfig(C.Config):
     """
 
 
-@plugin_registry.rebuild_on_registers
-@accelerator_registry.rebuild_on_registers
 class TrainerConfig(C.Config):
     # region Active Run Configuration
     id: str = C.Field(default_factory=lambda: TrainerConfig.generate_id())

{nshtrainer-1.0.0b33.dist-info → nshtrainer-1.0.0b36.dist-info}/METADATA

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

{nshtrainer-1.0.0b33.dist-info → nshtrainer-1.0.0b36.dist-info}/RECORD

@@ -1,65 +1,63 @@
 nshtrainer/.nshconfig.generated.json,sha256=yZd6cn1RhvNNJUgiUTRYut8ofZYvbulnpPG-rZIRhi4,106
-nshtrainer/__init__.py,sha256=52OB7QRlhrTCIdDecpT7yEZyZM1XvYxywhuORn1eKoY,814
+nshtrainer/__init__.py,sha256=g_moPnfQxSxFZX5NB9ILQQOJrt4RTRuiFt9N0STIpxM,874
 nshtrainer/_callback.py,sha256=tXQCDzS6CvMTuTY5lQSH5qZs1pXUi-gt9bQdpXMVdEs,12715
 nshtrainer/_checkpoint/metadata.py,sha256=PHy-54Cg-o3OtCffAqrVv6ZVMU7zhRo_-sZiSEEno1Y,5019
 nshtrainer/_checkpoint/saver.py,sha256=LOP8jjKF0Dw9x9H-BKrLMWlEp1XTan2DUK0zQUCWw5U,1360
 nshtrainer/_directory.py,sha256=p2uk1FnISFEpMqlDevKhoWhQsCEtvHUPg459K-86QA8,3053
 nshtrainer/_experimental/__init__.py,sha256=U4S_2y3zgLZVfMenHRaJFBW8yqh2mUBuI291LGQVOJ8,35
-nshtrainer/_hf_hub.py,sha256=xj91CEKcuvpaiipZbVZovZiU_fdsTkPXOkQ-3Xb-FhU,14183
-nshtrainer/callbacks/__init__.py,sha256=z2uDaUVzNXG79UOAYGpwC0hxk__LTbEBVhXJ3W-Srws,3911
-nshtrainer/callbacks/actsave.py,sha256=bnqS3y9sit1hfUvqPx-WWRug2OeTHPM0PDEJaex7f3Q,3776
-nshtrainer/callbacks/base.py,sha256=AFqvKNzH710xviYQ7X0hw0M7ETWRhBAWv3PN9WnrZw0,3608
-nshtrainer/callbacks/checkpoint/__init__.py,sha256=09cwgAJawTFLdzZwYhg_jVgbW-1d09hwjHdI-PQRck0,797
+nshtrainer/_hf_hub.py,sha256=4OsCbIITnZk_YLyoMrVyZ0SIN04FBxlC0ig2Et8UAdo,14287
+nshtrainer/callbacks/__init__.py,sha256=4giOYT8A709UOLRtQEt16QbOAFUHCjJ_aLB7ITTwXJI,3577
+nshtrainer/callbacks/actsave.py,sha256=NSXIIu62MNYe5gz479SMW33bdoKYoYtWtd_iTWFpKpc,3881
+nshtrainer/callbacks/base.py,sha256=Alaou1IHAIlMEM7g58d_02ozY2xWlshBN7fsw5Ee21s,3683
+nshtrainer/callbacks/checkpoint/__init__.py,sha256=l8tkHc83_mLiU0-wT09SWdRzwpm2ulbkLzcuCmuTwzE,620
 nshtrainer/callbacks/checkpoint/_base.py,sha256=ZVEUVl5kjCSSe69Q0rMUbKBNNUog0pxBwWkeyuxG2w0,6304
-nshtrainer/callbacks/checkpoint/best_checkpoint.py,sha256=hcvtpNvuyWxxP36XiGyFYwVsSwmHphKD-1IA95DinME,2602
-nshtrainer/callbacks/checkpoint/last_checkpoint.py,sha256=RKuFxFtaxUIM7xb0tOBoNSvkNjne2XBmJUEm_JE9lw4,1148
-nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py,sha256=5cCvLybQNSJy6Qm60oV2zxWQvGscM-0APXs-cApVwao,3464
-nshtrainer/callbacks/checkpoint/time_checkpoint.py,sha256=oom2e6-m_tKo0BKtjNDUanwvDQVOD9ZRu6cJ20APrqE,3561
-nshtrainer/callbacks/debug_flag.py,sha256=85IG9bFn2-bUUiEAgvfeteUhYTEuSyPRv9BYgRqe8mc,1687
-nshtrainer/callbacks/directory_setup.py,sha256=4Er9nOyNCMY9X27RzeW5z0Ky_OgnOBvxCF6xVsonMfg,2504
-nshtrainer/callbacks/early_stopping.py,sha256=8Ai5aZLQrM9zMRUSFxfJ2Jdb8OoBE8ZWAcGSyw8nM-k,4715
-nshtrainer/callbacks/ema.py,sha256=RbD6Mage7E2ke5-X4WcoOxMvNl4Z4_li2CnwmLFCCJc,12194
-nshtrainer/callbacks/finite_checks.py,sha256=iCiKQ5i9RckkzcPeCHzC3hkg3AlW3ESuWtFwoF2RG0Q,2124
-nshtrainer/callbacks/gradient_skipping.py,sha256=k5qNaNeileZ_5YFad4ssfLplMxMKeKFhPcY8-QVmLek,3464
+nshtrainer/callbacks/checkpoint/best_checkpoint.py,sha256=2CQuhPJ3Fi7lDw7z-J8kXXXuDU8-4HcU48oZxR49apk,2667
+nshtrainer/callbacks/checkpoint/last_checkpoint.py,sha256=MJcNB0biOebx2si2IBFaSUiVOSLSCZTzxB-RcEgO2gY,3482
+nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py,sha256=nljzETqkHwA-4g8mxaeFK5HxA8My0dlIPzIUscSMWyk,3525
+nshtrainer/callbacks/debug_flag.py,sha256=96fuP0C7C6dSs1GiMeUYzzs0X3Q4Pjt9JVWg3b75fU4,1748
+nshtrainer/callbacks/directory_setup.py,sha256=wPas_Ren8ANejogmIdKhqqgj4ulxz9AS_8xVIAfRXa0,2565
+nshtrainer/callbacks/early_stopping.py,sha256=EjzN-gD_Xd4YHZLkXsbi00g_4ti3RTMJEdHJ8GMeaFM,4776
+nshtrainer/callbacks/ema.py,sha256=dBFiUXG0xmyCw8-ayuSzJMKqSbepl6Ii5VIbhFlT5ug,12255
+nshtrainer/callbacks/finite_checks.py,sha256=3lZ3kEIjmYQfqTF0DcrgZ9_98ZLQhQj8usH7SgWst3o,2185
+nshtrainer/callbacks/gradient_skipping.py,sha256=8g7oC7PF0LTAEzwiNoaS5tWOnkjk_EB0QG3JdHkQ8ek,3523
 nshtrainer/callbacks/interval.py,sha256=UCzUzt3XCFVyQyCWL9lOrStkkxesvduNOYk8yMrGTTk,8116
-nshtrainer/callbacks/log_epoch.py,sha256=Wr-Ksxsynsqu_zyB_zoiPLjnWv-ksC3xPekY6iyN-P8,1396
-nshtrainer/callbacks/lr_monitor.py,sha256=IyFZoXaxJoTBSkdLu1iEZ1qI8_UFNJwafR_xTVPZXXU,1050
-nshtrainer/callbacks/norm_logging.py,sha256=C44Mvt73gqQEpCFd0j3qYg6NY7sL2jm3X1qJVY_XLfI,6329
-nshtrainer/callbacks/print_table.py,sha256=lS49Hz0OLcv3VPxEfLBguwe57y2nmKg0pMF6HJxuJio,2974
-nshtrainer/callbacks/rlp_sanity_checks.py,sha256=kWl2dYOXn2L8k6ub_012jNkqOxtyea1yr1qWRNG6UW4,9990
-nshtrainer/callbacks/shared_parameters.py,sha256=ggMI1krkqN7sGOrjK_I96IsTMYMXHoVQm7W90LZb9so,3015
-nshtrainer/callbacks/timer.py,sha256=BB-M7tV4QNYOwY_Su6j9P7IILxVRae_upmDq4qsxiao,4670
-nshtrainer/callbacks/wandb_upload_code.py,sha256=PTqNE1QB5U8NR5zhbiQZrmQuugX2UV7B12UdMpo9aV0,2353
-nshtrainer/callbacks/wandb_watch.py,sha256=tTTcFzxd2Ia9xu8tCogQ5CLJZBq1ne5JlpGVE75vKYs,2976
-nshtrainer/configs/__init__.py,sha256=eS3naq6EG1vCq28G2nAW1CqYFdsrh6ueBlzX_LazgUw,14159
+nshtrainer/callbacks/log_epoch.py,sha256=B5Dm8XVZwCzKUhUWfT_5PDdDac993191OsbcxxuSVJE,1457
+nshtrainer/callbacks/lr_monitor.py,sha256=qy_C0R40J0hBAukzBwng5FI2jJUpWuXOi5N6FU6ym3I,1210
+nshtrainer/callbacks/norm_logging.py,sha256=nVIDWe-ASl5zN830-ODR8QMCqI1ma-QPCIwoy0Wb-Nk,6390
+nshtrainer/callbacks/print_table.py,sha256=VaS4JgI963do79laXK4lUkFQx8v6aRSy22W0zyal_LA,3035
+nshtrainer/callbacks/rlp_sanity_checks.py,sha256=74BZvV2HLO__ucQXsLXb8eJLUZgRFUNJZ6TL9efMp74,10051
+nshtrainer/callbacks/shared_parameters.py,sha256=s94jJTAIbDGukYJu6l247QonVOCudGClU4t5kLt8XrY,3076
+nshtrainer/callbacks/timer.py,sha256=gDcw_K_ikf0bkVgxQ0cDhvvNvz6GLZVLcatuKfh0ORU,4731
+nshtrainer/callbacks/wandb_upload_code.py,sha256=shV7UtnXgY2bUlXdVrXiaDs0PNLlIt7TzNJkJPkzvzI,2414
+nshtrainer/callbacks/wandb_watch.py,sha256=VB14Dy5ZRXQ3di0fPv0K_DFJurLhroLPytnuwQBiJFg,3037
+nshtrainer/configs/__init__.py,sha256=OevZEZxb4H8imadSQXK9huqdYUF4SrJPfNU_2fpMBvI,14084
 nshtrainer/configs/_checkpoint/__init__.py,sha256=6s7Y68StboqscY2G4P_QG443jz5aiym5SjOogIljWLg,342
 nshtrainer/configs/_checkpoint/metadata/__init__.py,sha256=oOPfYkXTjKgm6pluGsG6V1TPyCEGjsQpHVL-LffSUFQ,290
 nshtrainer/configs/_directory/__init__.py,sha256=_oO7vM9DhzHSxtZcv86sTi7hZIptnK1gr-AP9mqQ370,386
-nshtrainer/configs/_hf_hub/__init__.py,sha256=VUgQnyEI2ekBxBIV15L09tKdrfGt7eWxnf30DiCLaso,416
-nshtrainer/configs/callbacks/__init__.py,sha256=APpF2jmafqbS4CoMmDvFADi0wdmXJ_BvFw4QnnQpok0,4353
-nshtrainer/configs/callbacks/actsave/__init__.py,sha256=JJg9d8iNGpO-9M1LsK4h1cu3NYWniyIyLQ4SauFCzOs,272
-nshtrainer/configs/callbacks/base/__init__.py,sha256=V694hzF_ubnA-hwTps30PeFbgDSm3I_UIMTnljM3_OI,176
-nshtrainer/configs/callbacks/checkpoint/__init__.py,sha256=zPUItxoYWrMT9i1TxOvVhIeTa0NEFg-nDE5FjHfkP-A,1564
+nshtrainer/configs/_hf_hub/__init__.py,sha256=ciFLbV-JV8SVzqo2SyythEuDMnk7gGfdIacB18QYnkY,511
+nshtrainer/configs/callbacks/__init__.py,sha256=jSWkbsdiu9vdGWTzqkDf-Bo9dXr9RengeNZLzWUhi7Y,4283
+nshtrainer/configs/callbacks/actsave/__init__.py,sha256=JvjSZtEoA28FC4u-QT3skQzBDVbN9eq07rn4u2ydW-E,377
+nshtrainer/configs/callbacks/base/__init__.py,sha256=wT3RhXttLyf6RFWCIvsoiXcPdfGx5W309WBI18AI5os,278
+nshtrainer/configs/callbacks/checkpoint/__init__.py,sha256=aGJ7vX14YamkMdwYAdPv6XrRnP0aZd5uZ5X0nSLc6IU,1475
 nshtrainer/configs/callbacks/checkpoint/_base/__init__.py,sha256=5jl6A5Gv6arZXmHV6lz5dQ8DL6PdJIfJqHLP4acClKQ,479
-nshtrainer/configs/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=_KTkF3_Yx0WiwOvRf2s1KRvod2dryeGJITVkx10YqBE,648
-nshtrainer/configs/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=sH4aJyovCeT6h4xz9r5WfVA0eviJur4zZt7R0hQsyAk,539
-nshtrainer/configs/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=iUoCrTJpvDGEYCYfNCpkficH6D7yq119hyxy1qeFSGU,410
-nshtrainer/configs/callbacks/checkpoint/time_checkpoint/__init__.py,sha256=ZQz5gXKSHz7xvnZbgU_bauNhAJDh9rblBA0ldkSsdwg,539
-nshtrainer/configs/callbacks/debug_flag/__init__.py,sha256=gPC3EAqzuyP2hAcCf3s09sPDe7q_02S1eUCWvRTNKrI,317
-nshtrainer/configs/callbacks/directory_setup/__init__.py,sha256=25551zMMctAkzcLEGBN7HSeQUIrBtbBq7whgPZjtepY,351
-nshtrainer/configs/callbacks/early_stopping/__init__.py,sha256=Q-hAAIcucLNriw5PXIshgBv7Yr5sMDk4GwjTIDdWBxo,434
-nshtrainer/configs/callbacks/ema/__init__.py,sha256=INIXgC7ZLmW0_ofa7vP51i2uaJKkmCiZtm75OaCrK7A,276
-nshtrainer/configs/callbacks/finite_checks/__init__.py,sha256=XlMOUyaBm417jhFTQhayexNzHd0t2FZyCYwbJ91qDcg,332
-nshtrainer/configs/callbacks/gradient_skipping/__init__.py,sha256=Jm5_pMk4qSo8bIJcxihEfOiGEBPPtbnJMA4ytYlryh4,361
-nshtrainer/configs/callbacks/log_epoch/__init__.py,sha256=0FR3-AvrS1M77BHlXrqOFstY9eXSjSNM5Eeyt-2HAyo,312
-nshtrainer/configs/callbacks/lr_monitor/__init__.py,sha256=a37cAuI8ZdG2e3RH4mBub_XIDWA9C8yTlYKciVFsdMA,323
-nshtrainer/configs/callbacks/norm_logging/__init__.py,sha256=jgfTvhhgy2p7eS4qbznIu1ohR_RMFiYdLfJJoyYrI7s,327
-nshtrainer/configs/callbacks/print_table/__init__.py,sha256=OD_Om09ZfyPLDLU-4MJURvrnrSe5kUaFhI1urEo8WYs,343
-nshtrainer/configs/callbacks/rlp_sanity_checks/__init__.py,sha256=SFI8w1Ne51Z3j3Vm1Ux1lRIvCQk5swVkHdJjFG2qSIk,358
-nshtrainer/configs/callbacks/shared_parameters/__init__.py,sha256=Ivef5jk3RMYQDeudEINxdzTnKfVZ_TsdtHlOTSl_EnQ,361
-nshtrainer/configs/callbacks/timer/__init__.py,sha256=RHOQoREp4NxS_AvKNdc0UuUlS0PnqCxxsuOz5D8h7iM,310
-nshtrainer/configs/callbacks/wandb_upload_code/__init__.py,sha256=WM9hCGFl2LXDUOgkIGaV3tkdnXnVBasrhIILjbIeFUo,358
-nshtrainer/configs/callbacks/wandb_watch/__init__.py,sha256=MW-ANrF529DxBhopovPjYEQ7nANX9ttd1K4_bJnKXks,322
+nshtrainer/configs/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=hL4WGBdo5_gtQuEGcRa3cWYMOSFdlNzkW-2Y3X3ZGTI,781
+nshtrainer/configs/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=SIRfz5QP30K4zzKw_1LZSSFr-3x-S3vc0vWL4ndyvjc,672
+nshtrainer/configs/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=VSkO0TYCAYy_9mQuOBoAND7D3Cg6w6nMCpqivQZLPcE,551
+nshtrainer/configs/callbacks/debug_flag/__init__.py,sha256=s_ifB-DbZjar0w11pr2oVAlcMTWWMnK_tCNilfswL04,425
+nshtrainer/configs/callbacks/directory_setup/__init__.py,sha256=e8GCRy2Alds3AXLwp4ieSGtn8S0YjmKJ5khOaQ0zKGs,464
+nshtrainer/configs/callbacks/early_stopping/__init__.py,sha256=m8N6H11PjqcWqXP5ZxWC8L4PHMUI6avYyN5rUNprjuQ,546
+nshtrainer/configs/callbacks/ema/__init__.py,sha256=DUJrbDD8wWX_s0_4dwKpT_IWKSVpBmhe4-1aELq7G6w,377
+nshtrainer/configs/callbacks/finite_checks/__init__.py,sha256=e-vx9Kn-noqw4wPvZw7fDMfb9Tsa6Duk0TIa8ZIgIIE,443
+nshtrainer/configs/callbacks/gradient_skipping/__init__.py,sha256=T3eVxxJfnYBrO9WfLiycn4TyWP4vaqJ57yp7Epkg7B4,485
+nshtrainer/configs/callbacks/log_epoch/__init__.py,sha256=IQ5owYYvyk7fiQP1QXYtncRRJrESuq3rRFhab-II2uE,419
+nshtrainer/configs/callbacks/lr_monitor/__init__.py,sha256=qejy1AnXNDHmsFuXRAXQQ5B0TcbKzvpaw-I4dv2AXIs,431
+nshtrainer/configs/callbacks/norm_logging/__init__.py,sha256=j2LrnYEbDGLwJR2lk-jmh-4J_iLEs2HNEoepvJSFLAg,437
+nshtrainer/configs/callbacks/print_table/__init__.py,sha256=t6fA_dBkUCszUXDJKEdnlBH4oEpfAQqcmAlatTFYIyQ,452
+nshtrainer/configs/callbacks/rlp_sanity_checks/__init__.py,sha256=dlP14Wh-w8zG_B4EtNmCIFzVMhf6bXCJ1O9cJWmEFnA,482
+nshtrainer/configs/callbacks/shared_parameters/__init__.py,sha256=AU7_bSnSRSlj16UTaFBe9QVUf0T0zKUmKCOBSL4xYmg,485
+nshtrainer/configs/callbacks/timer/__init__.py,sha256=cOUtbsl0_OhCO0fIcBfLuIF6FEGBHQu7AvQFzwVznWQ,413
+nshtrainer/configs/callbacks/wandb_upload_code/__init__.py,sha256=CJeCc9OCu5F39lWiY5aIc4WxQlgBvB-8cga6cQtw0GQ,482
+nshtrainer/configs/callbacks/wandb_watch/__init__.py,sha256=dzz1oavL1BwELE33xus45_avBEAZDeB6xtcb6CsOEos,431
 nshtrainer/configs/loggers/__init__.py,sha256=5wTekL79mQxit8f1K3AMllvb0mKertTzOKfC3gpE2Zk,1251
 nshtrainer/configs/loggers/_base/__init__.py,sha256=HxPPPePsEjlNuhnjsMgYIl0rwj_iqNKKOBTEk_zIOsM,169
 nshtrainer/configs/loggers/actsave/__init__.py,sha256=2lZQ4bpbjwd4MuUE_Z_PGbmQjjGtWCZUCtXqKO4dTSc,280
@@ -81,8 +79,8 @@ nshtrainer/configs/profiler/_base/__init__.py,sha256=ekYfPg-VDhCAFM5nJka2TxUYdRD
 nshtrainer/configs/profiler/advanced/__init__.py,sha256=-ThpUat16Ij_0avkMUVVA8wCWDG_q_tM7KQofnWQCtg,308
 nshtrainer/configs/profiler/pytorch/__init__.py,sha256=soAU1s2_Pa1na4gW8CK-iysJBO5M_7YeZC2_x40iEdg,294
 nshtrainer/configs/profiler/simple/__init__.py,sha256=3Wb11lPuFuyasq8xS1CZ4WLuBCLS_nVSQGVllvOOi0Y,289
-nshtrainer/configs/trainer/__init__.py,sha256=hKMI_2ve5zcsQys2DDQDv7OmshYsIG0uJlCLreVHpF0,7779
-nshtrainer/configs/trainer/_config/__init__.py,sha256=Xw6I_9tUemDbHncpjKHRqye_e1_OyubK_FJcvdcQ0yc,4020
+nshtrainer/configs/trainer/__init__.py,sha256=8Z4E1IeJHtDW8fpDxJkiC9CgDqKrTBIR5VMK1q4DYy4,7729
+nshtrainer/configs/trainer/_config/__init__.py,sha256=t72kmUn60UtjpD6H38XzKbEs50gU2dS1IH0u-RnHZ04,3666
 nshtrainer/configs/trainer/accelerator/__init__.py,sha256=3H6R3wlwbKL1TzDqGCChZk78-BcE2czLouo7Djiq3nA,898
 nshtrainer/configs/trainer/plugin/__init__.py,sha256=NkHQxMPkrtTtdIAO4dQUE9SWEcHRDB0yUXLkTjnl4dA,3332
 nshtrainer/configs/trainer/plugin/base/__init__.py,sha256=GuubcKrbXt4VjJRT8VpNUQqBtyuutre_CfkS0EWZ5_E,368
@@ -129,8 +127,8 @@ nshtrainer/profiler/_base.py,sha256=kFcSVn9gJuMwgDxbfyHh46CmEAIPZjxw3yjPbKgzvwA,
 nshtrainer/profiler/advanced.py,sha256=XrM3FX0ThCv5UwUrrH0l4Ow4LGAtpiBww2N8QAU5NOQ,1160
 nshtrainer/profiler/pytorch.py,sha256=8K37XvPnCApUpIK8tA2zNMFIaIiTLSoxKQoiyCPBm1Q,2757
 nshtrainer/profiler/simple.py,sha256=PimjqcU-JuS-8C0ZGHAdwCxgNLij4x0FH6WXsjBQzZs,1005
-nshtrainer/trainer/__init__.py,sha256=ggDHzIUbABezh4BjEwrxyWuXmuDBV-x4jv9gwXgVHU0,250
-nshtrainer/trainer/_config.py,sha256=0GgofvaWf5Vo9REXNJpTvpVVRlFExGTOzcOt4jwJXNk,34129
+nshtrainer/trainer/__init__.py,sha256=fQ7gQRlGWX-90TYT0rttkQyvXDCzo7DAvJgr-jX1zsY,316
+nshtrainer/trainer/_config.py,sha256=SPg3WXjF3ufhnr27sTHQLq23hdebnW6CTWa8AJkRG0A,32982
 nshtrainer/trainer/_runtime_callback.py,sha256=6F2Gq27Q8OFfN3RtdNC6QRA8ac0LC1hh4DUE3V5WgbI,4217
 nshtrainer/trainer/accelerator.py,sha256=Bqq-ry7DeCY4zw9_zBvTZiijpA-uUHrDjtbLV652m4M,2415
 nshtrainer/trainer/plugin/__init__.py,sha256=UM8f70Ml3RGpsXeQr1Yh1yBcxxjFNGFGgJbniCn_rws,366
@@ -153,6 +151,6 @@ nshtrainer/util/seed.py,sha256=diMV8iwBKN7Xxt5pELmui-gyqyT80_CZzomrWhNss0k,316
 nshtrainer/util/slurm.py,sha256=HflkP5iI_r4UHMyPjw9R4dD5AHsJUpcfJw5PLvGYBRM,1603
 nshtrainer/util/typed.py,sha256=Xt5fUU6zwLKSTLUdenovnKK0N8qUq89Kddz2_XeykVQ,164
 nshtrainer/util/typing_utils.py,sha256=MjY-CUX9R5Tzat-BlFnQjwl1PQ_W2yZQoXhkYHlJ_VA,442
-nshtrainer-1.0.0b33.dist-info/METADATA,sha256=HV4xY5ye1RhwlSttGP9OUTmRVzUUl7H8r_KZCsSN_YY,988
-nshtrainer-1.0.0b33.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-nshtrainer-1.0.0b33.dist-info/RECORD,,
+nshtrainer-1.0.0b36.dist-info/METADATA,sha256=R9O2SnflaNiDkxtoOPD_YFCXIgnEl8YjkhbEU5CbWHQ,988
+nshtrainer-1.0.0b36.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+nshtrainer-1.0.0b36.dist-info/RECORD,,

nshtrainer/callbacks/checkpoint/time_checkpoint.py

@@ -1,114 +0,0 @@
-from __future__ import annotations
-
-import logging
-import time
-from datetime import timedelta
-from pathlib import Path
-from typing import Any, Literal
-
-from lightning.pytorch import LightningModule, Trainer
-from typing_extensions import final, override
-
-from nshtrainer._checkpoint.metadata import CheckpointMetadata
-
-from ._base import BaseCheckpointCallbackConfig, CheckpointBase
-
-log = logging.getLogger(__name__)
-
-
-@final
-class TimeCheckpointCallbackConfig(BaseCheckpointCallbackConfig):
-    name: Literal["time_checkpoint"] = "time_checkpoint"
-
-    interval: timedelta = timedelta(hours=12)
-    """Time interval between checkpoints."""
-
-    @override
-    def create_checkpoint(self, trainer_config, dirpath):
-        return TimeCheckpointCallback(self, dirpath)
-
-
-@final
-class TimeCheckpointCallback(CheckpointBase[TimeCheckpointCallbackConfig]):
-    def __init__(self, config: TimeCheckpointCallbackConfig, dirpath: Path):
-        super().__init__(config, dirpath)
-        self.start_time = time.time()
-        self.last_checkpoint_time = self.start_time
-        self.interval_seconds = config.interval.total_seconds()
-
-    @override
-    def name(self):
-        return "time"
-
-    @override
-    def default_filename(self):
-        return "epoch{epoch}-step{step}-duration{train_duration}"
-
-    @override
-    def topk_sort_key(self, metadata: CheckpointMetadata):
-        return metadata.checkpoint_timestamp
-
-    @override
-    def topk_sort_reverse(self):
-        return True
-
-    def _should_checkpoint(self) -> bool:
-        current_time = time.time()
-        elapsed_time = current_time - self.last_checkpoint_time
-        return elapsed_time >= self.interval_seconds
-
-    def _format_duration(self, seconds: float) -> str:
-        """Format duration in seconds to a human-readable string."""
-        td = timedelta(seconds=int(seconds))
-        days = td.days
-        hours, remainder = divmod(td.seconds, 3600)
-        minutes, seconds = divmod(remainder, 60)
-
-        parts = []
-        if days > 0:
-            parts.append(f"{days}d")
-        if hours > 0:
-            parts.append(f"{hours}h")
-        if minutes > 0:
-            parts.append(f"{minutes}m")
-        if seconds > 0 or not parts:
-            parts.append(f"{seconds}s")
-
-        return "_".join(parts)
-
-    @override
-    def current_metrics(self, trainer: Trainer) -> dict[str, Any]:
-        metrics = super().current_metrics(trainer)
-        train_duration = time.time() - self.start_time
-        metrics["train_duration"] = self._format_duration(train_duration)
-        return metrics
-
-    @override
-    def on_train_batch_end(
-        self, trainer: Trainer, pl_module: LightningModule, *args, **kwargs
-    ):
-        if self._should_checkpoint():
-            self.save_checkpoints(trainer)
-            self.last_checkpoint_time = time.time()
-
-    @override
-    def state_dict(self) -> dict[str, Any]:
-        """Save the timer state for checkpoint resumption.
-
-        Returns:
-            Dictionary containing the start time and last checkpoint time.
-        """
-        return {
-            "start_time": self.start_time,
-            "last_checkpoint_time": self.last_checkpoint_time,
-        }
-
-    @override
-    def load_state_dict(self, state_dict: dict[str, Any]) -> None:
-        """Restore the timer state when resuming from a checkpoint.
-
-        Args:
-            state_dict: Dictionary containing the previously saved timer state.
-        """
-        self.start_time = state_dict["start_time"]
-        self.last_checkpoint_time = state_dict["last_checkpoint_time"]

nshtrainer/configs/callbacks/checkpoint/time_checkpoint/__init__.py

@@ -1,19 +0,0 @@
-from __future__ import annotations
-
-__codegen__ = True
-
-from nshtrainer.callbacks.checkpoint.time_checkpoint import (
-    BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig,
-)
-from nshtrainer.callbacks.checkpoint.time_checkpoint import (
-    CheckpointMetadata as CheckpointMetadata,
-)
-from nshtrainer.callbacks.checkpoint.time_checkpoint import (
-    TimeCheckpointCallbackConfig as TimeCheckpointCallbackConfig,
-)
-
-__all__ = [
-    "BaseCheckpointCallbackConfig",
-    "CheckpointMetadata",
-    "TimeCheckpointCallbackConfig",
-]