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

nshtrainer/trainer/_config.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from collections.abc import Iterable, Sequence
 from datetime import timedelta
@@ -30,14 +32,15 @@ from .._hf_hub import HuggingFaceHubConfig
 from ..callbacks import (
     BestCheckpointCallbackConfig,
     CallbackConfig,
-    EarlyStoppingConfig,
+    EarlyStoppingCallbackConfig,
     LastCheckpointCallbackConfig,
     OnExceptionCheckpointCallbackConfig,
 )
 from ..callbacks.base import CallbackConfigBase
 from ..callbacks.debug_flag import DebugFlagCallbackConfig
-from ..callbacks.rlp_sanity_checks import RLPSanityChecksConfig
-from ..callbacks.shared_parameters import SharedParametersConfig
+from ..callbacks.log_epoch import LogEpochCallbackConfig
+from ..callbacks.rlp_sanity_checks import RLPSanityChecksCallbackConfig
+from ..callbacks.shared_parameters import SharedParametersCallbackConfig
 from ..loggers import (
     CSVLoggerConfig,
     LoggerConfig,
@@ -65,7 +68,7 @@ class LoggingConfig(CallbackConfigBase):
 
     log_lr: bool | Literal["step", "epoch"] = True
     """If enabled, will register a `LearningRateMonitor` callback to log the learning rate to the logger."""
-    log_epoch: bool = True
+    log_epoch: LogEpochCallbackConfig | None = LogEpochCallbackConfig()
     """If enabled, will log the fractional epoch number to the logger."""
 
     actsave_logged_metrics: bool = False
@@ -136,9 +139,7 @@ class LoggingConfig(CallbackConfigBase):
             yield LearningRateMonitor(logging_interval=logging_interval)
 
         if self.log_epoch:
-            from ..callbacks.log_epoch import LogEpochCallback
-
-            yield LogEpochCallback()
+            yield from self.log_epoch.create_callbacks(root_config)
 
         for logger in self.loggers:
             if not logger or not isinstance(logger, CallbackConfigBase):
@@ -172,9 +173,9 @@ class OptimizationConfig(CallbackConfigBase):
 
     @override
     def create_callbacks(self, root_config):
-        from ..callbacks.norm_logging import NormLoggingConfig
+        from ..callbacks.norm_logging import NormLoggingCallbackConfig
 
-        yield from NormLoggingConfig(
+        yield from NormLoggingCallbackConfig(
             log_grad_norm=self.log_grad_norm,
             log_grad_norm_per_param=self.log_grad_norm_per_param,
             log_param_norm=self.log_param_norm,
@@ -564,8 +565,8 @@ class TrainerConfig(C.Config):
     reproducibility: ReproducibilityConfig = ReproducibilityConfig()
     """Reproducibility configuration options."""
 
-    reduce_lr_on_plateau_sanity_checking: RLPSanityChecksConfig | None = (
-        RLPSanityChecksConfig()
+    reduce_lr_on_plateau_sanity_checking: RLPSanityChecksCallbackConfig | None = (
+        RLPSanityChecksCallbackConfig()
     )
     """
     If enabled, will do some sanity checks if the `ReduceLROnPlateau` scheduler is used:
@@ -573,7 +574,7 @@ class TrainerConfig(C.Config):
         - If the `interval` is epoch, it makes sure that validation is called every `frequency` epochs.
     """
 
-    early_stopping: EarlyStoppingConfig | None = None
+    early_stopping: EarlyStoppingCallbackConfig | None = None
     """Early stopping configuration options."""
 
     profiler: ProfilerConfig | None = None
@@ -741,7 +742,9 @@ class TrainerConfig(C.Config):
     automatic selection based on the chosen accelerator. Default: ``"auto"``.
     """
 
-    shared_parameters: SharedParametersConfig | None = SharedParametersConfig()
+    shared_parameters: SharedParametersCallbackConfig | None = (
+        SharedParametersCallbackConfig()
+    )
     """If enabled, the model supports scaling the gradients of shared parameters that
     are registered in the self.shared_parameters list. This is useful for models that
     share parameters across multiple modules (e.g., in a GPT model) and want to

nshtrainer/trainer/_runtime_callback.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import datetime
 import logging
 import time

nshtrainer/trainer/checkpoint_connector.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from pathlib import Path
 from typing import TYPE_CHECKING, cast

nshtrainer/trainer/signal_connector.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 import os
 import platform

nshtrainer/trainer/trainer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 import os
 from collections.abc import Sequence

nshtrainer/util/_environment_info.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import getpass
 import importlib.metadata
 import inspect

nshtrainer/util/bf16.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import torch
 
 

nshtrainer/util/config/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from . import duration as duration
 from .dtype import DTypeConfig as DTypeConfig
 from .duration import DurationConfig as DurationConfig

nshtrainer/util/config/dtype.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import TYPE_CHECKING, Literal, TypeAlias
 
 import nshconfig as C

nshtrainer/util/config/duration.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import math
 from typing import Annotated, Literal
 

nshtrainer/util/environment.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 import os
 from contextlib import contextmanager

nshtrainer/util/path.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import hashlib
 import logging
 import os

nshtrainer/util/seed.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 
 import lightning.fabric.utilities.seed as LS

nshtrainer/util/slurm.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+
 class SlurmParseException(Exception):
     pass
 

nshtrainer/util/typed.py

@@ -1,2 +1,4 @@
+from __future__ import annotations
+
 from ..nn.module_dict import TypedModuleDict as TypedModuleDict
 from ..nn.module_list import TypedModuleList as TypedModuleList

nshtrainer/util/typing_utils.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import TYPE_CHECKING
 
 from typing_extensions import TypeVar

{nshtrainer-0.41.1.dist-info → nshtrainer-0.43.0.dist-info}/METADATA

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

nshtrainer-0.43.0.dist-info/RECORD

@@ -0,0 +1,162 @@
+nshtrainer/__init__.py,sha256=KgAwRuVmetgBOeEfG_09sJ6JO-F3LH5QsBRDiVdNm6o,678
+nshtrainer/_callback.py,sha256=qPvUK3cZZyin9WNWXdKLd4RxqMbaCzyrqnhXdQMIRYg,1049
+nshtrainer/_checkpoint/loader.py,sha256=ViXEr7OPyekNE4FgbBpMkS6ilY7QGLWKtiyTZaqQunc,14194
+nshtrainer/_checkpoint/metadata.py,sha256=4WCXT4nosS4_P1xurZ4dhadSP7iVLB8qyMvCq8rsu8A,4773
+nshtrainer/_checkpoint/saver.py,sha256=LOP8jjKF0Dw9x9H-BKrLMWlEp1XTan2DUK0zQUCWw5U,1360
+nshtrainer/_directory.py,sha256=p2uk1FnISFEpMqlDevKhoWhQsCEtvHUPg459K-86QA8,3053
+nshtrainer/_experimental/__init__.py,sha256=1eAD28upA3EEHsxKKkNXMdl78IML55CsMtiDYF0tsJc,117
+nshtrainer/_hf_hub.py,sha256=JO6Hup2BM0GV6s65LvzAOFtT1u7zmO-mFcAyKaOHYnk,14141
+nshtrainer/callbacks/__init__.py,sha256=KCdmHAXyqiWqdjbGuHnuiqSTZFQgMtZcghRNCV9uuAY,3815
+nshtrainer/callbacks/_throughput_monitor_callback.py,sha256=5-3pk1S2NzJBfdAhlxVLVsoJsewXP0W47udmRz34r_0,23220
+nshtrainer/callbacks/actsave.py,sha256=VY85oImVcwIasvBmF5qX8l_K1lZ88yJTW3zJYqK_0zw,4077
+nshtrainer/callbacks/base.py,sha256=AWuwxpqg-l2PNQjGbxlOwRxRPItLp84t6n-2a6eoIZk,3544
+nshtrainer/callbacks/checkpoint/__init__.py,sha256=SlXrYBF6Sb1ToH5O6P7HcbJBedqJGsEdISlouWVRD3Y,604
+nshtrainer/callbacks/checkpoint/_base.py,sha256=wcrRcFIc59OWBLSGYrKNc6XjkVwAs0ono4P2zFg8keQ,6251
+nshtrainer/callbacks/checkpoint/best_checkpoint.py,sha256=SHsZDRGOzrEsAWLWlbYB2-bLB-E6vPbDXwNSOOacs1E,2589
+nshtrainer/callbacks/checkpoint/last_checkpoint.py,sha256=79Eh1JX7XePp-Jix-ERe-4-tYJE55P-crF41GhWz5pU,1154
+nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py,sha256=1Y_76X2SpqBDbILfJKW7pZK0OWLyUhQ1y23qEOUwEE8,3446
+nshtrainer/callbacks/debug_flag.py,sha256=rUQZUfUURHS7Nti9STyzM5ad9kRv_nZjISiR4UU38hU,2047
+nshtrainer/callbacks/directory_setup.py,sha256=k8vpaR_T2TRgZLlB6GAt68LYIzzp-mBAwy4YgVw6ATM,2681
+nshtrainer/callbacks/early_stopping.py,sha256=xM2V7-1JXb7aULQx5EZOOR7gJGlFBdu38_FXSDeADIc,4706
+nshtrainer/callbacks/ema.py,sha256=A64Ah9OUaEi4pj1u6af4Ds5-X0MuW1wu9oVBUdN7qzQ,12191
+nshtrainer/callbacks/finite_checks.py,sha256=EImZPPdS_0VcbVb_REXSYqYl1twPRygreDVeBt_uPAY,2121
+nshtrainer/callbacks/gradient_skipping.py,sha256=lNLhYN4MTZNhI5DER0SLk8NLXrAZw8uPlYAtYVKW5K0,3461
+nshtrainer/callbacks/interval.py,sha256=UCzUzt3XCFVyQyCWL9lOrStkkxesvduNOYk8yMrGTTk,8116
+nshtrainer/callbacks/log_epoch.py,sha256=3bBA0wJMd2sDe_0nYDznOYhcBqdaTz6ua7HlY0AVN6Y,1393
+nshtrainer/callbacks/norm_logging.py,sha256=M81V_DVrGDCcep1UN0fqn1ZC5ZH7biSXjiSdNjZUvf4,6326
+nshtrainer/callbacks/print_table.py,sha256=1o0dMG9CwpBdEGQeG61uwc9QBA692Fy_evwwQJoFq9g,2881
+nshtrainer/callbacks/rlp_sanity_checks.py,sha256=4UnHffbkgT1bCHRP9-KVdbQ2CGGbGvg5zEcUhASW0FU,9987
+nshtrainer/callbacks/shared_parameters.py,sha256=mxa0BseYBj_R-sFZpe-omrIgRlwcQl0766A8v61gD9M,2974
+nshtrainer/callbacks/throughput_monitor.py,sha256=kfM--rVUAm1x9InNMuRcm0jkGN6ecRxSG2w7-NDh3XA,1874
+nshtrainer/callbacks/timer.py,sha256=cf4iHdL-W7w6AddJ2L7VmlOqhCULRaYQY-zuKn6hruQ,4667
+nshtrainer/callbacks/wandb_upload_code.py,sha256=ZQgcm10QuP6wWPIJcLOraoZOPZkl3jfnmEiJsFQ24XA,2350
+nshtrainer/callbacks/wandb_watch.py,sha256=x-DGaDcXpBBuN09QumeHmvGCbTm_IBEesn9yPSyhmTo,2973
+nshtrainer/config/__init__.py,sha256=3fzl30I9Dj9ycFwhGhnx4oFyDiGMUFBhJKWKCVpDdb0,22695
+nshtrainer/config/_checkpoint/loader/__init__.py,sha256=LscuP8J9zSUfjWxKl-uryJzIjQO4_kRkOcFTp868KIk,2399
+nshtrainer/config/_checkpoint/metadata/__init__.py,sha256=X9KxpcoHQbJp6-MTGvp4pct-MYHaHcl82s9yqZ5KiSk,867
+nshtrainer/config/_directory/__init__.py,sha256=w9xYSZRgWq4lp5iBYa3tX5MWtQe5tpYdNSVxv9-nM9k,1029
+nshtrainer/config/_hf_hub/__init__.py,sha256=fXSaOru7AIgIHXKLA2jnPiXyivUdECu7Y5jN5qmTrqk,1063
+nshtrainer/config/callbacks/__init__.py,sha256=K-8apQVhVGh-07wNIkLRqmpWunXvgatf72qnWcnx3WI,7338
+nshtrainer/config/callbacks/actsave/__init__.py,sha256=AkVWS9vCcDJFpPUpyc7i9cjaFZU2kKxDyFDqakMZA-E,809
+nshtrainer/config/callbacks/base/__init__.py,sha256=OdtHDMkYC_ioCEAkg7bSQi3o7e2t5WHPcFjavXdfdTA,602
+nshtrainer/config/callbacks/checkpoint/__init__.py,sha256=TB7AE7X24S3g0xiO4aD0Uzg-pDLlTI7damfR2o83o9k,2700
+nshtrainer/config/callbacks/checkpoint/_base/__init__.py,sha256=rYQ7hPaqv606sY42qjF-9mwBZZdH0T-rorpFRz2IWbA,1255
+nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=dkGFUTuDSppQ__xYggqBPw4a08p0DaiSX0y1dvncHHY,1637
+nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=jrX5HHyoxGPnzlsSq_a2OV6ENXMSRc7aK-LU31818k4,1355
+nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=unh9ao4mJ64aY0k0JXCdVBYyk1_cU5lYqlKOH0nQbO0,1069
+nshtrainer/config/callbacks/debug_flag/__init__.py,sha256=RQiUwWeaR8pcXKGYP5oZnnYALLtdRallgLI6VYB96Ao,908
+nshtrainer/config/callbacks/directory_setup/__init__.py,sha256=dixNsxyeVulEcMqBZKdolZDcl3R8jXhubMQ_Y8PhJX8,965
+nshtrainer/config/callbacks/early_stopping/__init__.py,sha256=NYs90lfyKD_amIM_-u2eJ8CH6UYKLfR-4udmJu18huA,1198
+nshtrainer/config/callbacks/ema/__init__.py,sha256=WpSohYCGplaRYDeXPPoAn-CnafGYw3mRo89gZQp4FnA,809
+nshtrainer/config/callbacks/finite_checks/__init__.py,sha256=VuLUoTMLgQPa3R9mxkkZRIyAeXQyVdC9eNVUhMzmsek,949
+nshtrainer/config/callbacks/gradient_skipping/__init__.py,sha256=bphb_C91L6RPhUWcVIX_IxBxgFvGkuZD8TVDsPqFywU,981
+nshtrainer/config/callbacks/norm_logging/__init__.py,sha256=x654QELJyLdxlDyt7YrSzVn1-hFNbphkiK4kn2kFbr4,941
+nshtrainer/config/callbacks/print_table/__init__.py,sha256=Z_7QFur2da8V14v5w0YpJldXj-P-ypp59Uta-MMXS5A,961
+nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py,sha256=RiTOGnct1THb8jI_reSL12HELMQZPanveJUsLSWkuTc,977
+nshtrainer/config/callbacks/shared_parameters/__init__.py,sha256=xNlohJ4UFV-Ojwbh3p0G8o0onbZeTEjpOe_boXUnQmw,981
+nshtrainer/config/callbacks/throughput_monitor/__init__.py,sha256=9pxqThqXOi_RZiDWaHGVN4cLbyxdAZG0toEdKdJVr2M,957
+nshtrainer/config/callbacks/timer/__init__.py,sha256=j1_vVpn7U9A8t6EqkV0_1OFAJZLxeV2rZBOZB-bX4tE,892
+nshtrainer/config/callbacks/wandb_upload_code/__init__.py,sha256=0QEHuBYLjyEvCOzTg6tDAKyHnYyzc4k94PtmOBUsisM,977
+nshtrainer/config/callbacks/wandb_watch/__init__.py,sha256=LE7PAsy5r1PttMDsKFYkmGNxgDrKV7ThhcQj5zqrW3k,933
+nshtrainer/config/loggers/__init__.py,sha256=ZLq0EzVXymNryGr_yLrvrzTZF07cRVkRpGD6WxGeIu4,2290
+nshtrainer/config/loggers/_base/__init__.py,sha256=CFo8APn-VRiUI19lcSV97icCJrKefQ1wAFuZTWFn_fQ,562
+nshtrainer/config/loggers/csv/__init__.py,sha256=AgKd0slcqBzfvTX-OkpF_y-EK8zZdR4XX_Ye01DPvT4,755
+nshtrainer/config/loggers/tensorboard/__init__.py,sha256=sw2lvaAJLxFQze5fxY-WJfJHHGZ2RX3VpZ-oqW9a_Ck,896
+nshtrainer/config/loggers/wandb/__init__.py,sha256=xjoA8voAjFRqraj0O6Aktx8mazh5iHAHqJXYsc5SZKo,1602
+nshtrainer/config/lr_scheduler/__init__.py,sha256=f2Rkja13ciy7T2DWitjoVr-J9AuZ1oB6oohBUZ8Y25s,2149
+nshtrainer/config/lr_scheduler/_base/__init__.py,sha256=u9bOJOs_BrZLTg-aUZpfsNDvtE6957KsCgfoV2IuvjA,639
+nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py,sha256=7l9zhLnkX-QlfMdmv0x0_HPxo7SplH6NkeQNRThEphM,1341
+nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py,sha256=CN_4VtFGk7QzhOpRAo-sQ8QqGLg_n4cPxkH6gE6dzDQ,1265
+nshtrainer/config/metrics/__init__.py,sha256=W_nAvsNNCdwGKbYMer93GO2fei9LJacmlRiRkKbLjQI,568
+nshtrainer/config/metrics/_config/__init__.py,sha256=SX9sHbKqloCw5jGv1zS-JmF3M5AaXpdpxJZNE0SPTes,550
+nshtrainer/config/model/__init__.py,sha256=0Hk6gJbeNqiQpk3A729724fBFKZ_xYkDWadlEwg-K4Y,1561
+nshtrainer/config/model/base/__init__.py,sha256=wmksFIT_909P3YMRKwMzvOVtUHaKQgEVH9XkZ4vJY14,735
+nshtrainer/config/model/config/__init__.py,sha256=Aubl0ypBxiNA67qg6833XBhvCN2EKHkjRv87B4Chq3o,1531
+nshtrainer/config/model/mixins/logger/__init__.py,sha256=shkRA1JaWq4GVCO1DkqTq4v8m7e4o2THyFhfGmAobJQ,550
+nshtrainer/config/nn/__init__.py,sha256=WGKJ4Y5sPWv9VsWeYsgfepIJFvfHp25WXYJEkQ19Ldg,3990
+nshtrainer/config/nn/mlp/__init__.py,sha256=C6dft6w162_J3YggaC5JmpnEjPUcbiw4lSu9FFB0SPU,934
+nshtrainer/config/nn/nonlinearity/__init__.py,sha256=PLYSXOdzd3yub56QQbiQukr3eTagdrgyLIaVtKAEwe8,4797
+nshtrainer/config/optimizer/__init__.py,sha256=F-x3t2IsG0gOMzu226LvDhINiXBTxKw8CiOuUtIzMVs,936
+nshtrainer/config/profiler/__init__.py,sha256=Df7w-VGwDBOYJDYPL9U_5OwwXOjrqQkYb3QFb6o8yyg,1521
+nshtrainer/config/profiler/_base/__init__.py,sha256=smDYCplrI5B38XJcNZ462ZeTo9l1ieLXCxRxf-FJEUY,602
+nshtrainer/config/profiler/advanced/__init__.py,sha256=VCCvbzhEeOcdZ0Unvk_anAcmbQuGogTQhK_bXs5RG9U,892
+nshtrainer/config/profiler/pytorch/__init__.py,sha256=I6mQfAvu1J6bTVci3o3sK8TdOFDBW1cZ1JTWXN8M7Xg,884
+nshtrainer/config/profiler/simple/__init__.py,sha256=jZEke3iaciGrl4k81hGOWMXFZ1CAI9mYAZsvexy8o-c,859
+nshtrainer/config/runner/__init__.py,sha256=rvwxfAHjrYRxhV-LoMi6uN0HtHinAts3qeofeLcQcx4,524
+nshtrainer/config/trainer/_config/__init__.py,sha256=2WDcrUabrlh_dfvp6BDA4_OBmpXSNb-MskER222AQsk,6634
+nshtrainer/config/trainer/checkpoint_connector/__init__.py,sha256=pSu79zOFFWvqjI3SkHWl13H8ZNJFTc6a5J1r2KnfUKM,667
+nshtrainer/config/util/_environment_info/__init__.py,sha256=mSzPR3mfvMx0ZQrnn2DDyNuCTc2inLKYmHJfEZfsoNk,3730
+nshtrainer/config/util/config/__init__.py,sha256=L-H6GvRWkHZRFnfD-mlYi5cu3Hbtrq4Ih6w7T0TZ5EU,1162
+nshtrainer/config/util/config/dtype/__init__.py,sha256=NCXMVO-EUz3JvPmlDci72O9ZeqpCANs0ieeoXAu9Ljw,550
+nshtrainer/config/util/config/duration/__init__.py,sha256=8llT1MCKQpsdNldN5h5Wo0GjUuRn28Sxw2FTXTNKBpM,1060
+nshtrainer/data/__init__.py,sha256=K4i3Tw4g9EOK2zlMMbidi99y0SyI4w8P7_XUf1n42Ts,260
+nshtrainer/data/balanced_batch_sampler.py,sha256=r1cBKRXKHD8E1Ax6tj-FUbE-z1qpbO58mQ9VrK9uLnc,5481
+nshtrainer/data/datamodule.py,sha256=roT4gfadPpvD-b63_IbzuxmlOjOQI-g3nZJpPvuQXZ0,149
+nshtrainer/data/transform.py,sha256=qd0lIocO59Fk_m90xyOHgFezbymd1mRwly8nbYIfHGc,2263
+nshtrainer/ll/__init__.py,sha256=eSRSqJLHbm7SFaVF5-b-wTwm3oTriRAhpo_ChhwamlM,2513
+nshtrainer/ll/_experimental.py,sha256=lhaCxbUEoZXBn7yd_K0vPch6dg1_TVvSB458tVge2g4,89
+nshtrainer/ll/actsave.py,sha256=zNj16Xv_-holuVI14csp2a-YZjJlONXJIWPnbNsi1VU,245
+nshtrainer/ll/callbacks.py,sha256=WmS0qATlKJYTq5AiY5OY6axmhC5FpV_rtW69wHNcd1k,85
+nshtrainer/ll/config.py,sha256=LFdF3X69loEPQexZ7zkUHRjqyrXpNtaEnerEEn71r-E,181
+nshtrainer/ll/data.py,sha256=TXawppvLXy_N1-CyheHvPuZp_2ZMJQKrmKGgHofNrw4,80
+nshtrainer/ll/log.py,sha256=STn2bIhMOKTQuuS0Jvv01DQhQYpdV4_gQIqjcSNLRts,176
+nshtrainer/ll/lr_scheduler.py,sha256=dis2zS_BqlWn6A-LAkxLbGj1OLjP5OEs9kIEywKXXCE,88
+nshtrainer/ll/model.py,sha256=usuP2kVczhOdGhTxA4_t6fhwrXl-wTslbqh96StprUY,1032
+nshtrainer/ll/nn.py,sha256=G2JjuGSJ_dtK-tQUvtwvMPgwQx7QQ7pA32miksY33iY,78
+nshtrainer/ll/optimizer.py,sha256=m5D3btw3RNGRY37NMrI4yKyglmGBkjnAE37u078keQo,85
+nshtrainer/ll/runner.py,sha256=s424AeQ25NweAxIHyLkSb7CynhhlyQBeZs4rvPK1_pE,153
+nshtrainer/ll/snapshot.py,sha256=IMGBd0xrOppZPXdcgVe2rXBfMSU6QYgtApFM8tVyTqM,123
+nshtrainer/ll/snoop.py,sha256=zPkIoSsZJElG0kWmXnk_YFyQDo1EVH31NcL5wRLxub8,72
+nshtrainer/ll/trainer.py,sha256=iZ-fAsNWAvFwaydXJZrx3_hsQYq10TgZHIJ4sHVfMIo,83
+nshtrainer/ll/typecheck.py,sha256=i8SgaP7C-cWELMWqXGS-7vVegPFUgAF1nj0H_xCV4us,98
+nshtrainer/ll/util.py,sha256=dsdhTO3tGg5OD7NUlmIUyuUbiMupdSUnZfohr3zKb1M,80
+nshtrainer/loggers/__init__.py,sha256=d8aB7NEc57vOFi1Cl7S_Adwk77CenAfT6GbpjunTlDM,484
+nshtrainer/loggers/_base.py,sha256=eTeLKKOl8s_vOhU4_JNwcWEciabDi-elAV7K7PMUOyY,741
+nshtrainer/loggers/csv.py,sha256=ahbQj0uquf6WiIPZV1hUPxz8FuQwvfPww9BHyBvzkJo,1088
+nshtrainer/loggers/tensorboard.py,sha256=yP9hebra1NeNiEAB7WrlJzh1AGYHLOvC0vP_Ku82sfE,2252
+nshtrainer/loggers/wandb.py,sha256=VTPy1HtAgnzmvTYYuHJ7cL1exOL95nyd93jlm65OY5o,6746
+nshtrainer/lr_scheduler/__init__.py,sha256=o7eDaj-KOetQvEwMjGKpcQofs1jimqUzHbJ5YM5cOMY,772
+nshtrainer/lr_scheduler/_base.py,sha256=WnD7_fvkXckMbIc-ZkcbLm9umf3vm1NRJLWA5fiC8Bk,3695
+nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=g9F5EXXW4LRoOXP6WY3V-5cYnDLjlRKkJdh7YrOrWnc,5406
+nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=phDJgjVGFUGurWFIECcKOxK-EdBeTcDxIlcSl9X-988,2818
+nshtrainer/metrics/__init__.py,sha256=Nqkn_jsDf3n5WtfMcnaaEftYjIIT2b-S7rmsB1MOMkU,86
+nshtrainer/metrics/_config.py,sha256=kR9OJLZXIaZpG8UpHG3EQwOL36HB8yPk6afYjoIM0XM,1324
+nshtrainer/model/__init__.py,sha256=bKgE221066M2QlmLMjMCyh0qhjpMWWaWTAfW9HLKgAE,297
+nshtrainer/model/base.py,sha256=lUYrkbBYdbTmOr94HnvS_ZLQlaWRWxEH3URdQ00sqFs,18645
+nshtrainer/model/config.py,sha256=ez1XgWdyZTxh6ETWE51p7CEH5o_owVgSIqUlYXEv56M,6997
+nshtrainer/model/mixins/callback.py,sha256=USsff85bvcrOOZPyYbs6AxfcqkO7RyCbzEfbp5nFANs,2455
+nshtrainer/model/mixins/logger.py,sha256=b6aH9A9SeQOetBNapVVfD3wCrJgSKWJu6s8biwazbBc,5327
+nshtrainer/nn/__init__.py,sha256=sANhrZpeN5syLKOsmXMwhaFl2SBFPWcLaEe1EH22TWQ,1463
+nshtrainer/nn/mlp.py,sha256=2W8bzE96DzCMzGm6WPiPhNFQfhqaoG3GXPn_oKBnlUM,5988
+nshtrainer/nn/module_dict.py,sha256=9plb8aQUx5TUEPhX5jI9u8LrpTeKe7jZAHi8iIqcN8w,2365
+nshtrainer/nn/module_list.py,sha256=UB43pcwD_3nUke_DyLQt-iXKhWdKM6Zjm84lRC1hPYA,1755
+nshtrainer/nn/nonlinearity.py,sha256=mp5XvXRHURB6jwuZ0YyTj5ZoHJYNJNgO2aLtUY1D-2Y,6114
+nshtrainer/optimizer.py,sha256=73UWfuTU4moDQNSKI8VSQSDc6_i7mBtemxvhBkM-fC4,1563
+nshtrainer/profiler/__init__.py,sha256=X6E4cbIRaXHCDgQtT3LFPw5xIRBhqnJ03SVFbsD-Yd8,516
+nshtrainer/profiler/_base.py,sha256=uPlH3BIBMK22upQQe2cLmp_jl20WV1P9SaEg2s_sEy0,933
+nshtrainer/profiler/advanced.py,sha256=I1pJckyt4sF_y-KAnoCC1cfZMBQyOj3v-Hv13m175kE,1148
+nshtrainer/profiler/pytorch.py,sha256=mILwyMTPpqMkHxiijd5GTatw50mo435ZMYfbuDcNKCs,2745
+nshtrainer/profiler/simple.py,sha256=Z86mhUbQ04RlgHo6EhJ-9PcROebjVObk6ybK-zEQ4QY,993
+nshtrainer/runner.py,sha256=UIaTPhgmkM_1oF0Qv_av9VMr8DgJJ4OFASz3n8VT19E,3189
+nshtrainer/scripts/find_packages.py,sha256=-oNnSNPp3pujCVgManW_LFlJcnvhrtHgvUJ4W88-6-o,1460
+nshtrainer/trainer/__init__.py,sha256=tvb6xb3lMfXGagnDcUKT0k-QZQFshUWdpErFGZOjQfo,76
+nshtrainer/trainer/_config.py,sha256=vexsMRDbGuHFUFbNASw-t5lrcq7qM506Dy-6Zwska3k,29399
+nshtrainer/trainer/_runtime_callback.py,sha256=T3epaj1YeIN0R8CS2cg5HNJIB21TyaD_PVNNOPJ6nJs,4200
+nshtrainer/trainer/checkpoint_connector.py,sha256=6ekND0oiIx7nVQBBHSlI_pqmtGifrCcnIe_6Z23AyCo,2383
+nshtrainer/trainer/signal_connector.py,sha256=YMJf6vTnW0JcnBkuYikm9x_9XscaokrCEzCn4THOGao,10776
+nshtrainer/trainer/trainer.py,sha256=RLeFU0sP_-NFWJ62GYOQuGP2369Oetj52e5yfaQsi2M,16875
+nshtrainer/util/_environment_info.py,sha256=B6IABVgmZVEp2x-zhf0eqjC9_CUuJoqyyzIH44yGDuw,24249
+nshtrainer/util/_useful_types.py,sha256=dwZokFkIe7M5i2GR3nQ9A1lhGw06DMAFfH5atyquqSA,8000
+nshtrainer/util/bf16.py,sha256=9QhHZCkYSfYpIcxwAMoXyuh2yTSHBzT-EdLQB297jEs,762
+nshtrainer/util/config/__init__.py,sha256=Z39JJufSb61Lhn2GfVcv3eFW_eorOrN9-9llDWlnZZM,272
+nshtrainer/util/config/dtype.py,sha256=LBndiDd00c29ULFrmG1R7yPvPO4fxHqFYQGacAx8BPA,2001
+nshtrainer/util/config/duration.py,sha256=mM-UfU_HvhXwW33TYEDg0x58n80tnle2e6VaWtxZTjk,764
+nshtrainer/util/environment.py,sha256=s-B5nY0cKYXdFMdNYumvC_xxacMATiI4DvV2gUDu20k,4195
+nshtrainer/util/path.py,sha256=mfuMRk29YpmwdRsyyf99VT0n_T9caM99SS7VfA6oIHE,3501
+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=5EDJ2C_DEZauWwWOyNWBCVOJJL3znHPkayJcRq3SHPk,421
+nshtrainer-0.43.0.dist-info/METADATA,sha256=z2RdjeMxEuaeRNNJW_Dv27YJ2WwzV4WWYn0KEyBpAks,916
+nshtrainer-0.43.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-0.43.0.dist-info/RECORD,,

nshtrainer-0.41.1.dist-info/RECORD

@@ -1,162 +0,0 @@
-nshtrainer/__init__.py,sha256=8hx3uqzroMYOQM7S4U3Eznw7MHF3YDqUkO1MDQeC4cM,642
-nshtrainer/_callback.py,sha256=A1zLsTy4b_wOYnInLLXGSRdHzT2yNa6mPEql-ozm0u0,1013
-nshtrainer/_checkpoint/loader.py,sha256=5vjg-OFChXJjgiOVv8vnV8nwTscfdDtEdxQRz6uPfDE,14158
-nshtrainer/_checkpoint/metadata.py,sha256=5D4PgKodzhLsmQvuF3xxkH49epKaegxi4wh_ImDTtns,4737
-nshtrainer/_checkpoint/saver.py,sha256=xuA9iDpc4wAs4ieYW1eJoJicMn6e5xCeCIEKfVOCvrY,1324
-nshtrainer/_directory.py,sha256=RjnW6vKTeKlz2vQWT3cG0Jje5BkFXA7HpUubDhcSiq4,2993
-nshtrainer/_experimental/__init__.py,sha256=pEXPyI184UuDHvfh4p9Kg9nQZQZI41e4_HvNd4BK-yg,81
-nshtrainer/_hf_hub.py,sha256=woRRy9bzAQ8ueoH903CzYGjQYZ0Cp7Hw5uv8qgLnZ5o,14105
-nshtrainer/callbacks/__init__.py,sha256=y4QKDiI4IykcHncDOE-OemFFQqegxyeoD_3v9i2OwFw,3248
-nshtrainer/callbacks/_throughput_monitor_callback.py,sha256=aJo_11rc4lo0IYOd-kHmPDtzdC4ctgXyRudkRJqH4m4,23184
-nshtrainer/callbacks/actsave.py,sha256=qbnaKts4_dvjPeAaPtv7Ds12_vEWzaHUfg_--49NB9I,4041
-nshtrainer/callbacks/base.py,sha256=NpjeKmonJ1Kaz5_39XSn3LlDwvbGjk6WV8BpHSNCvI4,3508
-nshtrainer/callbacks/checkpoint/__init__.py,sha256=g-3zIthupERKqWZQw-A_busQPaPRkto6iHBV-M7nK1Y,527
-nshtrainer/callbacks/checkpoint/_base.py,sha256=X6844ceZ-EMz9UCVLJnA1d-ej2AkkRH90v2hy2zPDkc,6215
-nshtrainer/callbacks/checkpoint/best_checkpoint.py,sha256=MOC5jLlQnnc61yuiVc2-O4NeYclnkXroOPDV9-2zh5w,2553
-nshtrainer/callbacks/checkpoint/last_checkpoint.py,sha256=CWWv0cSwQ1VAX26N7hAyMxbNCk26Keh39oQguBEK5To,1102
-nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py,sha256=ctT88EGT22_t_6tr5r7Sfo43cuve6XeroBnBYRMPOus,3372
-nshtrainer/callbacks/debug_flag.py,sha256=T7rkY9hYQ_-PsPo2XiQ4eVZ9bBsTd2knpZWctCbjxXc,2011
-nshtrainer/callbacks/directory_setup.py,sha256=c0uY0oTqLcQ3egInHO7G6BeQQgk_xvOLoHH8FR-9U0U,2629
-nshtrainer/callbacks/early_stopping.py,sha256=upCNtjkXmwGPwLJmDZacXdBzySG2i_sGIK9QjnaE6tU,4638
-nshtrainer/callbacks/ema.py,sha256=8-WHmKFP3VfnzMviJaIFmVD9xHPqIPmq9NRF5xdu3c8,12131
-nshtrainer/callbacks/finite_checks.py,sha256=gJC_RUr3ais3FJI0uB6wUZnDdE3WRwCix3ppA3PwQXA,2077
-nshtrainer/callbacks/gradient_skipping.py,sha256=EBNkANDnD3BTszWjnG-jwY8FEj-iRqhE3e1x5LQF6M8,3393
-nshtrainer/callbacks/interval.py,sha256=smz5Zl8cN6X6yHKVsMRS2e3SEkzRCP3LvwE1ONvLfaw,8080
-nshtrainer/callbacks/log_epoch.py,sha256=fTa_K_Y8A7g09630cG4YkDE6AzSMPkjb9bpPm4gtqos,1120
-nshtrainer/callbacks/norm_logging.py,sha256=T2psu8mYsw9iahPKT6aUPjkGrZ4TIzm6_UUUmE09GJs,6274
-nshtrainer/callbacks/print_table.py,sha256=_FdAHhqylWGk4Z0c2FrLFeiMA4jhfA_beZRK_BHpzmE,2837
-nshtrainer/callbacks/rlp_sanity_checks.py,sha256=c30G9jAu42QLLIS5LnusdSnI3wqyIHgOUFDRcKESuNI,9935
-nshtrainer/callbacks/shared_parameters.py,sha256=fqlDweFDXPV_bfcAWpRgaJIad9i5AehYDtuJjDtUum4,2922
-nshtrainer/callbacks/throughput_monitor.py,sha256=H_ocXErZxUO3dxFk8Tx_VQdpI9E_Ztvqof5WtFevLyQ,1838
-nshtrainer/callbacks/timer.py,sha256=quS79oYClDUvQxJkNWmDMe0hwRUkkREgTgqzVrnom50,4607
-nshtrainer/callbacks/wandb_upload_code.py,sha256=OWG4UkL2SfW6oj6AGRXeBJsZmgsqeHLW2Fj8Jm4ga3I,2298
-nshtrainer/callbacks/wandb_watch.py,sha256=Y6SEXfIx3kDDQbI5zpP53BVq0FBLJbLd3RJsiHZk1-Y,2921
-nshtrainer/config/__init__.py,sha256=y-K4iv44DZkpV0cIQglNLmxHbxVBVRO5DLq9E34M-HI,7808
-nshtrainer/config/_checkpoint/loader/__init__.py,sha256=wpbjL4GsMfYmb0PM3yj6Qry-d7aKxM4yZ7s7Qp9Qn3g,803
-nshtrainer/config/_checkpoint/metadata/__init__.py,sha256=3iSkHEmKTw0nCaGfzWroUHN6h6naHfLOi7EySCH8yKw,282
-nshtrainer/config/_directory/__init__.py,sha256=1Qf7Q0OOwwCQmPNm1VM7vFjuaL25B-Wnzcz5ZtuSESI,325
-nshtrainer/config/_hf_hub/__init__.py,sha256=7qz4dzkjoOocv9biRTKCVsQI-GgSE5dAOft2s7tGgsw,358
-nshtrainer/config/callbacks/__init__.py,sha256=u9bcllwXYEGxSWFKcDDP16qQzPx_P2eGYqKDRdKMc-c,2698
-nshtrainer/config/callbacks/actsave/__init__.py,sha256=LB2dnKpCUTy2_15hpQgZ77Y_9GmALlxK2V01CpyCIQ8,268
-nshtrainer/config/callbacks/base/__init__.py,sha256=V4fzP1q_LpXPoGV5Ve5mXFRscNJ0AQkr-4k5iJDWBxU,193
-nshtrainer/config/callbacks/checkpoint/__init__.py,sha256=d3Y0I3aomhvyVva6PF5nraJb3aM9rb5gDZNB4_7z7mg,1017
-nshtrainer/config/callbacks/checkpoint/_base/__init__.py,sha256=7iZ3aTak0NOiLI5s5tOpY-PJyq6h8wEybf0U0VeuBNk,407
-nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=rST4HhhOhti10QZjWBu0CkJErjNTI8cP44e0hQtX_j8,546
-nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=gMxD2quXg7VCa1vkzjSSjegtBXgQJ6lpvq-OaLeicFI,457
-nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=2WHicpnirFzydPA9wsRDXqN6xwT2W7eVqCdDNqtX9xs,366
-nshtrainer/config/callbacks/debug_flag/__init__.py,sha256=hiiZvFDLjUDpH3Xpk9IerIRNTmHhycJVSEMPPrrqUi8,294
-nshtrainer/config/callbacks/directory_setup/__init__.py,sha256=NswK3ywE-oikMUzYgJ3-bXK1TPcf5OB5Ce2mslO4gLY,298
-nshtrainer/config/callbacks/early_stopping/__init__.py,sha256=VpkNZ7z0s_iQP_IGsUNk-s6c-hSiqLydHFQ1NkqCjjM,371
-nshtrainer/config/callbacks/ema/__init__.py,sha256=id-m1X5Q-Rkw9HJo4J72V59EjedOxlHiqLW4GWa8J34,252
-nshtrainer/config/callbacks/finite_checks/__init__.py,sha256=utfprFQgrhrk3CFaN6twRu35ryJzD_L2qwFYN5VmJcM,290
-nshtrainer/config/callbacks/gradient_skipping/__init__.py,sha256=mrPYNjVQMgKfNM8rStmNAeATOTus1685D8LB4JYbszs,306
-nshtrainer/config/callbacks/norm_logging/__init__.py,sha256=w2l08YgAbxy3EjDaZO3f0s8NgURXFZnW5xy1n8JFUKU,286
-nshtrainer/config/callbacks/print_table/__init__.py,sha256=Ne6O1JbSnPwCj4H225FBbJnHRoIR4rBnIG2pogLeHZU,296
-nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py,sha256=-VVBo8P7JgI0J2JbehOqi4BnrDxwmRrBkZBxQSsDNKE,304
-nshtrainer/config/callbacks/shared_parameters/__init__.py,sha256=UH6U4_xaBwpk-otAi2k4un9sNNDOykJ0l26vkq4tX1c,306
-nshtrainer/config/callbacks/throughput_monitor/__init__.py,sha256=Eafoki1Ja4qEgiGmCxLVimQtB5ObxVcqRDfdYEjoTX0,310
-nshtrainer/config/callbacks/timer/__init__.py,sha256=pl0ZiZZdeJI9im65mkVQ1NoYP_ZgV-vHGXLtecjLAXM,270
-nshtrainer/config/callbacks/wandb_upload_code/__init__.py,sha256=QGwSTtQZFMJnVIGZsMwbAMRHy_qvHB-S4ysEvT47VGU,304
-nshtrainer/config/callbacks/wandb_watch/__init__.py,sha256=1qdlXUkzfuLCr6_x9Yor1rnIrxai2Ev5swSl3T0Tjfo,282
-nshtrainer/config/loggers/__init__.py,sha256=c5bs11U2DCHUeqrQsPPzhZVOu_OHoZrFVXdLsewOMcY,820
-nshtrainer/config/loggers/_base/__init__.py,sha256=EAQOKsfk-c9hIPZ6ohjv1r9WpDCVD5mWVGrfu5ST5OY,188
-nshtrainer/config/loggers/csv/__init__.py,sha256=YnUMUPucEHaBSiXErhAlP_GOTckb_g530ttoCy-rJdk,256
-nshtrainer/config/loggers/tensorboard/__init__.py,sha256=nq18V6CMd-kz7mvgxEgNUqztNrU8dbvdNbwZBGJcwe4,288
-nshtrainer/config/loggers/wandb/__init__.py,sha256=nY6plVAb0f3NBenq3LYHnXy_XauqGQWnGw7Z8ik_Ug0,500
-nshtrainer/config/lr_scheduler/__init__.py,sha256=VNAFT6BbQQ4EPY7AfCMTCba0-K5wr0cp5c4FwK67Aj4,803
-nshtrainer/config/lr_scheduler/_base/__init__.py,sha256=RO6gI-P9SOgH17m3ycRns9IgiFda6oIOAV4Xqa9-0og,203
-nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py,sha256=uzisQaTWTVxZL3mPQJBdJy16hrfiCUAzP4hgRdWDcow,450
-nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py,sha256=yI7Hpa4CNX4Ys7QIq_emLiWbr6tghf-oVjVjIq5k1hE,412
-nshtrainer/config/metrics/__init__.py,sha256=P6goxe9RhwQZx8cC2gZ1dfoofLOTII-O-Y1Gas3km9k,207
-nshtrainer/config/metrics/_config/__init__.py,sha256=lTdHS7SFD-8VFkalFqNltPrhWxbWFEovt1XPmC8cusc,182
-nshtrainer/config/model/__init__.py,sha256=u8pWp7RaDqG9kAlKVnMFzOCNyqisOmk5ouH1FQ_xY-4,589
-nshtrainer/config/model/base/__init__.py,sha256=zL1ZUA8OaeGSyA1Beyf_UmNjEWHayHdX5KNxFCkxMcg,246
-nshtrainer/config/model/config/__init__.py,sha256=rD2CeNx3prtnxv5xNluMiDbSotRpyJnHZhNxVG5vuaM,530
-nshtrainer/config/model/mixins/logger/__init__.py,sha256=phExLK3R5R5okUA-DcekxoWZgU1oEgIyzdOG9jL3TUY,182
-nshtrainer/config/nn/__init__.py,sha256=AutesBnPzVh1CWXe8yYXwr_ZxQ4KGdUOAd1BTp1tjw0,1456
-nshtrainer/config/nn/mlp/__init__.py,sha256=F5eSCQM8lqTA7fGDFeQLX1vyS0BAg6Br1SFbwI3z2E4,317
-nshtrainer/config/nn/nonlinearity/__init__.py,sha256=lhx-Mitn6BZ-wdDuE70OT9mWNycPs_CoWLDmuLQxZdo,1534
-nshtrainer/config/optimizer/__init__.py,sha256=LxnWMttr594voY2iewD-QgpLJ0LYQG2h3fCxiJPFd5o,318
-nshtrainer/config/profiler/__init__.py,sha256=cKvQwqdrc4htiYsL5ahAFdoJw0IGqnK_Nf99suDYstw,617
-nshtrainer/config/profiler/_base/__init__.py,sha256=Pi5HckHRSywRsvn9Cd84tqRwZD5UH8cDOS4_f8zM8P0,193
-nshtrainer/config/profiler/advanced/__init__.py,sha256=gXn8K98CUCYy4m_Efv0yIuOnvxBvKdmZc_7r6tmsKec,286
-nshtrainer/config/profiler/pytorch/__init__.py,sha256=UO_FQhue8hpHS7_WsdMDw_dWTtDi_gFvnr1TBWk2u7U,282
-nshtrainer/config/profiler/simple/__init__.py,sha256=Wbec--_xC1oUkmiA7pZw4f5emuMTEOn-n6vJ3Jk0Jas,278
-nshtrainer/config/runner/__init__.py,sha256=NAvbAyrdKy8u_uRaNCHP7eJz9X2anXQAB75TPYyDbe0,169
-nshtrainer/config/trainer/_config/__init__.py,sha256=MG9d1ayQWlV54xoBRQlBUHxG2ihdBqmH1b0VtdFpdSk,2140
-nshtrainer/config/trainer/checkpoint_connector/__init__.py,sha256=CM2wmQ8KirlMqx4EYnttL_M4fjdT9MhYQjvhqZRyyRM,217
-nshtrainer/config/util/_environment_info/__init__.py,sha256=e11YIxaFL_6o7CiYIn80mYfaPSUmvKJ2RfftAhjTzeg,1237
-nshtrainer/config/util/config/__init__.py,sha256=5PQPUvyx_ycS-H6NdGebFS2W_6cFozLsrjeI-ITluVQ,434
-nshtrainer/config/util/config/dtype/__init__.py,sha256=fzOWOq-WMhNw-ZOoU3VnaPAKPvVy4IfL2L0MEzdTRg0,182
-nshtrainer/config/util/config/duration/__init__.py,sha256=zOA-9xum4WhS1IWfP1-YP__u1Axzc5js-9Xz015QFQ0,335
-nshtrainer/data/__init__.py,sha256=fnigdEtNcB8DJqz4MHvcI6UwOZICFzWItskY78lLTMA,224
-nshtrainer/data/balanced_batch_sampler.py,sha256=ybMJF-CguaZ17fLEweZ5suaGOiHOMEm3Bn8rQfGTzGQ,5445
-nshtrainer/data/datamodule.py,sha256=Dk06LdcbK13sCLQWacdCoXU_CHWZxoKEZ_xpAg7nZQg,113
-nshtrainer/data/transform.py,sha256=6SNs3_TpNpfhcwTwvPKyEJ3opM1OT7LmMEYQNHKgRl8,2227
-nshtrainer/ll/__init__.py,sha256=L-aTi1V1bbvnZjOro8NvI393zbHQSFR9movWSRK9Mds,2477
-nshtrainer/ll/_experimental.py,sha256=oBQCKOEVYoxuUU9eLb-Fg2B2mzZD7SA0zfAO6lmWZ88,53
-nshtrainer/ll/actsave.py,sha256=2lbiseSrjcwFT6AiyLNWarTWl1bnzliVWlu1iOfnP30,209
-nshtrainer/ll/callbacks.py,sha256=AxyUmc8aGRSjx6WwwgXYCmdJ73rwLuEAEH0AGRosojQ,49
-nshtrainer/ll/config.py,sha256=fKumJf42HY2FITX1QUM1OTXkYD6U2np2ciyd4PFRPZ8,145
-nshtrainer/ll/data.py,sha256=zRG0FRje-jtSHximVzkHIHzpwsyQxpHCoACFihNKLPM,44
-nshtrainer/ll/log.py,sha256=d4BB3TyM8imK65EXOiOeUTF0zFM1ropbe7Vq3DeB0xU,140
-nshtrainer/ll/lr_scheduler.py,sha256=7xjhN6L69BCUzFhcy33NtMtPuCzHiB611zVWFg92lQ0,52
-nshtrainer/ll/model.py,sha256=Cw8Vq8IUL6YU1fTUcOIZsXcNJ3XyKgQY4YENIsL9H7c,996
-nshtrainer/ll/nn.py,sha256=8qiRDFwojIxkB7-LtNWk4mLL2tJbaskHYofDsOIHiNg,42
-nshtrainer/ll/optimizer.py,sha256=3T-VZtT73jVvwCNJGDjgGEbzs-1LFTzMQH-SB_58mSo,49
-nshtrainer/ll/runner.py,sha256=B0m5VEhNKIjF1aFmqPkonkQxDoRL2jeHZGsV3zwhSVE,117
-nshtrainer/ll/snapshot.py,sha256=EdzvbE6IyHcNpIvkF1lWj3pBigfRF9jQXgOlbgBavYs,87
-nshtrainer/ll/snoop.py,sha256=hG9VCdm8mIZytHLZgUKRSoWqg55rVvUpBAH-OSiwgvI,36
-nshtrainer/ll/trainer.py,sha256=hkn2xPtrSPQ7LqQhbyAKuMfNyHdhqB9bDPvgRCK8oJM,47
-nshtrainer/ll/typecheck.py,sha256=ryV1Tzcf7hJ4I19H1oQVkikU9spmRk8jyIKQZ5UF7pQ,62
-nshtrainer/ll/util.py,sha256=PQGu5Ff1raizxjXdm2rFFu4Mo816dBIYmItjkJX1_qk,44
-nshtrainer/loggers/__init__.py,sha256=C_xk0A3_qKbNdTmzK85AgjRHFD3w-jPRS2ig-iPhfEk,448
-nshtrainer/loggers/_base.py,sha256=xiZKEK0ALJkcqf4OpVNRY0QbZsamR_WR7x7m_68YHXQ,705
-nshtrainer/loggers/csv.py,sha256=D_lYyd94bZ8jAgnRo-ARtFgVcInaD9zktxtsUD9RWCI,1052
-nshtrainer/loggers/tensorboard.py,sha256=wL2amRSdP68zbslZvBeM0ZQBnjF3hIKsz-_lBbdomaM,2216
-nshtrainer/loggers/wandb.py,sha256=GMlEzkFesgS3qgVgYqATPHJ3rvs9etwFlk3HbidOyRQ,6662
-nshtrainer/lr_scheduler/__init__.py,sha256=uEvgaFAs-4s_bAEMaildy0GT6OvgpgOEKTuzqutESHE,736
-nshtrainer/lr_scheduler/_base.py,sha256=7xOIuxQ86YHbFWG5a3gX46emQj1WN_LaY4-i0Q1TDBg,3659
-nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=E8LW78uuby7bIsoLPpcF1bmNK4lSko-r3qPL-vuHWXQ,5370
-nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=h76oTHYpMxauV_l6lviya5DW-WKArwxxf7ZQizhmbCw,2782
-nshtrainer/metrics/__init__.py,sha256=ObLIELGguIEcUpRsUkqh1ltrvZii6vglTpJGrPvoy00,50
-nshtrainer/metrics/_config.py,sha256=jgRBfDAQLFTW7AiUY7CRtdfts6CR6keeuqm0FFMWCzQ,1288
-nshtrainer/model/__init__.py,sha256=2i_VEy6u_Y1LUGKljHXWeekvhnUcanZM2QyaaBM1Bmw,261
-nshtrainer/model/base.py,sha256=NasbYZJBuEly6Hm9t9HVZk-CUHmy4T7p1v-Ye981XA4,18609
-nshtrainer/model/config.py,sha256=gyP5uuDfl4EbkmFlqnpNm49D0PgJWK9S6UOFGGaAAtQ,6961
-nshtrainer/model/mixins/callback.py,sha256=rbe8P22iEjPkH1df6rfEo3Txw7EwSz6Dkm0TWO_AysM,2419
-nshtrainer/model/mixins/logger.py,sha256=xOymSTofukEYZGkGojXsMEO__ZlBI5lIPZVmlotMEX8,5291
-nshtrainer/nn/__init__.py,sha256=0QPFl02a71WZQjLMGOlFNMmsYP5aa1q3eABHmnWH58Q,1427
-nshtrainer/nn/mlp.py,sha256=V0FrScpIUdg_IgIO8GMtIsGEtmHjwF14i2IWxmZrsqg,5952
-nshtrainer/nn/module_dict.py,sha256=NOY0B6WDTnktyWH4GthsprMQo0bpehC-hCq9SfD8paE,2329
-nshtrainer/nn/module_list.py,sha256=bbR62bnQM4No6HtmBllijOhLwDtWwmcPcO2DklPbsbc,1719
-nshtrainer/nn/nonlinearity.py,sha256=4sYE4MN5zojc-go1k0PYtqssVRuXrM7D4tbpIXp5K-E,6078
-nshtrainer/optimizer.py,sha256=kuJEA1pvB3y1FcsfhAoOJujVqEZqFHlmYO8GW6JeA1g,1527
-nshtrainer/profiler/__init__.py,sha256=cJ_wAm8j3Bz6cKgNQ_9gQLZz9nddKW53VE81UKg8l8g,480
-nshtrainer/profiler/_base.py,sha256=YF5lsJBIl9qts9GLW5Z62JuYeo4SnIArhlFwTGkfTb4,897
-nshtrainer/profiler/advanced.py,sha256=44asloha0aGUW8YwjQt3lm3ve8H-N6mM4QgseUSLT30,1112
-nshtrainer/profiler/pytorch.py,sha256=tGeRvoPP5ulWX2RkfXrQvMBoki1T95dpz5p8mwyon1I,2709
-nshtrainer/profiler/simple.py,sha256=MbMfsJvligd0mtGiltxJ0T8MQVDP9T9BzQZFwswl66Y,957
-nshtrainer/runner.py,sha256=llW5PThXfzlu8ym1waFVzED1RGI91OHqN3j02sepoZo,3153
-nshtrainer/scripts/find_packages.py,sha256=ixYivZobumyyGsf2B9oYMLyLTRcBzY_vUv-u3bNW-hs,1424
-nshtrainer/trainer/__init__.py,sha256=P2rmr8oBVTHk-HJHYPcUwWqDEArMbPR4_rPpATbWK3E,40
-nshtrainer/trainer/_config.py,sha256=YqpGb4RodkUg87TVE5WBSc4CQkUF0z3qDRdil1HRxoM,29198
-nshtrainer/trainer/_runtime_callback.py,sha256=sd2cUdRJG-UCdQr9ruZvEYpNGNF1t2W2fuxwwVlQD9E,4164
-nshtrainer/trainer/checkpoint_connector.py,sha256=r0ir4xYSdf_jebM0x09qaO6nJsvsiRQDyM0fs80ppOQ,2347
-nshtrainer/trainer/signal_connector.py,sha256=2EzkVktlasl8PgWAKNLDZRUMY__gRlDy1HdinAU-tfU,10740
-nshtrainer/trainer/trainer.py,sha256=8T4LB31ygXXS3DECkvD2uqgElAxkulacYvZyL_-imJs,16839
-nshtrainer/util/_environment_info.py,sha256=CFUUZYjXhBLWGc0jtPNOaZgYMueUDEHpEaWFA1f3GoY,24213
-nshtrainer/util/_useful_types.py,sha256=dwZokFkIe7M5i2GR3nQ9A1lhGw06DMAFfH5atyquqSA,8000
-nshtrainer/util/bf16.py,sha256=VUnIG6aA4XtZscZc_dxv5ln_jlEbdU3eMFwDb5SEWSI,726
-nshtrainer/util/config/__init__.py,sha256=o8fwPf_dctE_7CAkT0wNOBkvmxnzYzXeHpLedrZLt54,236
-nshtrainer/util/config/dtype.py,sha256=JtYjrcBFNBlziJnLAE6QS0QV4PUXhGspYH1hNFrB3ks,1965
-nshtrainer/util/config/duration.py,sha256=pgIKQ88Dg8y1YAKUvUsNWu9hc9O79kdYBfgmC3a_-kQ,728
-nshtrainer/util/environment.py,sha256=AeW_kLl-N70wmb6L_JLz1wRj0kA70xs6RCmc9iUqczE,4159
-nshtrainer/util/path.py,sha256=BmBzqtxGu7w2gTYkxQevdBffu_Crm1dH3TURoqqjR6M,3465
-nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
-nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
-nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
-nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
-nshtrainer-0.41.1.dist-info/METADATA,sha256=fZxCkOyqGGEG8YHHzAvjZWK6V70wvORAAptibUvhkBw,916
-nshtrainer-0.41.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-0.41.1.dist-info/RECORD,,