nshtrainer 1.0.0b55__py3-none-any.whl → 1.1.0__py3-none-any.whl

nshtrainer/configs/trainer/plugin/base/__init__.py

@@ -2,12 +2,10 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from nshtrainer.trainer.plugin.base import PluginConfig as PluginConfig
 from nshtrainer.trainer.plugin.base import PluginConfigBase as PluginConfigBase
 from nshtrainer.trainer.plugin.base import plugin_registry as plugin_registry
 
 __all__ = [
-    "PluginConfig",
     "PluginConfigBase",
     "plugin_registry",
 ]

nshtrainer/configs/trainer/plugin/environment/__init__.py

@@ -2,7 +2,6 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from nshtrainer.trainer.plugin.environment import DTypeConfig as DTypeConfig
 from nshtrainer.trainer.plugin.environment import (
     KubeflowEnvironmentPlugin as KubeflowEnvironmentPlugin,
 )
@@ -28,7 +27,6 @@ from nshtrainer.trainer.plugin.environment import (
 from nshtrainer.trainer.plugin.environment import plugin_registry as plugin_registry
 
 __all__ = [
-    "DTypeConfig",
     "KubeflowEnvironmentPlugin",
     "LSFEnvironmentPlugin",
     "LightningEnvironmentPlugin",

nshtrainer/configs/trainer/plugin/io/__init__.py

@@ -5,7 +5,6 @@ __codegen__ = True
 from nshtrainer.trainer.plugin.io import (
     AsyncCheckpointIOPlugin as AsyncCheckpointIOPlugin,
 )
-from nshtrainer.trainer.plugin.io import PluginConfig as PluginConfig
 from nshtrainer.trainer.plugin.io import PluginConfigBase as PluginConfigBase
 from nshtrainer.trainer.plugin.io import (
     TorchCheckpointIOPlugin as TorchCheckpointIOPlugin,
@@ -15,7 +14,6 @@ from nshtrainer.trainer.plugin.io import plugin_registry as plugin_registry
 
 __all__ = [
     "AsyncCheckpointIOPlugin",
-    "PluginConfig",
     "PluginConfigBase",
     "TorchCheckpointIOPlugin",
     "XLACheckpointIOPlugin",

nshtrainer/trainer/plugin/__init__.py

@@ -1,10 +1,18 @@
 from __future__ import annotations
 
+from typing import Annotated
+
+from typing_extensions import TypeAliasType
+
 from . import environment as environment
 from . import io as io
 from . import layer_sync as layer_sync
 from . import precision as precision
 from .base import Plugin as Plugin
-from .base import PluginConfig as PluginConfig
 from .base import PluginConfigBase as PluginConfigBase
 from .base import plugin_registry as plugin_registry
+
+PluginConfig = TypeAliasType(
+    "PluginConfig",
+    Annotated[PluginConfigBase, plugin_registry.DynamicResolution()],
+)

nshtrainer/trainer/plugin/base.py

@@ -1,8 +1,7 @@
 from __future__ import annotations
 
-import logging
 from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING, Annotated
+from typing import TYPE_CHECKING
 
 import nshconfig as C
 from lightning.fabric.plugins import CheckpointIO, ClusterEnvironment
@@ -12,7 +11,6 @@ from typing_extensions import TypeAliasType
 
 if TYPE_CHECKING:
     from .._config import TrainerConfig
-log = logging.getLogger(__name__)
 
 
 Plugin = TypeAliasType(
@@ -26,8 +24,3 @@ class PluginConfigBase(C.Config, ABC):
 
 
 plugin_registry = C.Registry(PluginConfigBase, discriminator="name")
-
-PluginConfig = TypeAliasType(
-    "PluginConfig",
-    Annotated[PluginConfigBase, plugin_registry.DynamicResolution()],
-)

nshtrainer/trainer/plugin/environment.py

@@ -1,27 +1,25 @@
 from __future__ import annotations
 
 import signal
-from typing import Any, Literal
+from typing import Literal
 
-from lightning.pytorch.plugins.environments import ClusterEnvironment
-from typing_extensions import override
+from typing_extensions import TypeAliasType, override
 
-from ...util.config.dtype import DTypeConfig
 from .base import PluginConfigBase, plugin_registry
 
 
 @plugin_registry.register
 class KubeflowEnvironmentPlugin(PluginConfigBase):
-    name: Literal["kubeflow_environment"] = "kubeflow_environment"
-
     """Environment for distributed training using the PyTorchJob operator from Kubeflow.
 
     This environment, unlike others, does not get auto-detected and needs to be passed
     to the Fabric/Trainer constructor manually.
     """
 
+    name: Literal["kubeflow_environment"] = "kubeflow_environment"
+
     @override
-    def create_plugin(self, trainer_config) -> ClusterEnvironment:
+    def create_plugin(self, trainer_config):
         from lightning.fabric.plugins.environments.kubeflow import KubeflowEnvironment
 
         return KubeflowEnvironment()
@@ -29,8 +27,6 @@ class KubeflowEnvironmentPlugin(PluginConfigBase):
 
 @plugin_registry.register
 class LightningEnvironmentPlugin(PluginConfigBase):
-    name: Literal["lightning_environment"] = "lightning_environment"
-
     """The default environment used by Lightning for a single node or free cluster (not managed).
 
     There are two modes the Lightning environment can operate with:
@@ -40,8 +36,10 @@ class LightningEnvironmentPlugin(PluginConfigBase):
        The appropriate environment variables need to be set, and at minimum `LOCAL_RANK`.
     """
 
+    name: Literal["lightning_environment"] = "lightning_environment"
+
     @override
-    def create_plugin(self, trainer_config) -> ClusterEnvironment:
+    def create_plugin(self, trainer_config):
         from lightning.fabric.plugins.environments.lightning import LightningEnvironment
 
         return LightningEnvironment()
@@ -49,16 +47,16 @@ class LightningEnvironmentPlugin(PluginConfigBase):
 
 @plugin_registry.register
 class LSFEnvironmentPlugin(PluginConfigBase):
-    name: Literal["lsf_environment"] = "lsf_environment"
-
     """An environment for running on clusters managed by the LSF resource manager.
 
     It is expected that any execution using this ClusterEnvironment was executed
     using the Job Step Manager i.e. `jsrun`.
     """
 
+    name: Literal["lsf_environment"] = "lsf_environment"
+
     @override
-    def create_plugin(self, trainer_config) -> ClusterEnvironment:
+    def create_plugin(self, trainer_config):
         from lightning.fabric.plugins.environments.lsf import LSFEnvironment
 
         return LSFEnvironment()
@@ -66,48 +64,108 @@ class LSFEnvironmentPlugin(PluginConfigBase):
 
 @plugin_registry.register
 class MPIEnvironmentPlugin(PluginConfigBase):
-    name: Literal["mpi_environment"] = "mpi_environment"
-
     """An environment for running on clusters with processes created through MPI.
 
     Requires the installation of the `mpi4py` package.
     """
 
+    name: Literal["mpi_environment"] = "mpi_environment"
+
     @override
-    def create_plugin(self, trainer_config) -> ClusterEnvironment:
+    def create_plugin(self, trainer_config):
         from lightning.fabric.plugins.environments.mpi import MPIEnvironment
 
         return MPIEnvironment()
 
 
+SignalAlias = TypeAliasType(
+    "SignalAlias",
+    Literal[
+        "SIGABRT",
+        "SIGFPE",
+        "SIGILL",
+        "SIGINT",
+        "SIGSEGV",
+        "SIGTERM",
+        "SIGBREAK",
+        "CTRL_C_EVENT",
+        "CTRL_BREAK_EVENT",
+        "SIGALRM",
+        "SIGBUS",
+        "SIGCHLD",
+        "SIGCONT",
+        "SIGHUP",
+        "SIGIO",
+        "SIGIOT",
+        "SIGKILL",
+        "SIGPIPE",
+        "SIGPROF",
+        "SIGQUIT",
+        "SIGSTOP",
+        "SIGSYS",
+        "SIGTRAP",
+        "SIGTSTP",
+        "SIGTTIN",
+        "SIGTTOU",
+        "SIGURG",
+        "SIGUSR1",
+        "SIGUSR2",
+        "SIGVTALRM",
+        "SIGWINCH",
+        "SIGXCPU",
+        "SIGXFSZ",
+        "SIGEMT",
+        "SIGINFO",
+        "SIGCLD",
+        "SIGPOLL",
+        "SIGPWR",
+        "SIGRTMAX",
+        "SIGRTMIN",
+        "SIGSTKFLT",
+    ],
+)
+
+
 @plugin_registry.register
 class SLURMEnvironmentPlugin(PluginConfigBase):
+    """An environment for running on clusters managed by the SLURM resource manager."""
+
     name: Literal["slurm_environment"] = "slurm_environment"
 
     auto_requeue: bool = True
     """Whether automatic job resubmission is enabled or not."""
 
-    requeue_signal: signal.Signals | None = None
+    requeue_signal: SignalAlias | None = None
     """The signal that SLURM will send to indicate that the job should be requeued."""
 
     @override
-    def create_plugin(self, trainer_config) -> ClusterEnvironment:
+    def create_plugin(self, trainer_config):
         from lightning.fabric.plugins.environments.slurm import SLURMEnvironment
 
+        requeue_signal = None
+        if self.requeue_signal is not None:
+            try:
+                requeue_signal = signal.Signals[self.requeue_signal]
+            except KeyError:
+                raise ValueError(
+                    f"Invalid signal name: {self.requeue_signal}. "
+                    "Please provide a valid signal name from the signal module."
+                )
+
         return SLURMEnvironment(
             auto_requeue=self.auto_requeue,
-            requeue_signal=self.requeue_signal,
+            requeue_signal=requeue_signal,
         )
 
 
 @plugin_registry.register
 class TorchElasticEnvironmentPlugin(PluginConfigBase):
-    name: Literal["torchelastic_environment"] = "torchelastic_environment"
-
     """Environment for fault-tolerant and elastic training with torchelastic."""
 
+    name: Literal["torchelastic_environment"] = "torchelastic_environment"
+
     @override
-    def create_plugin(self, trainer_config) -> ClusterEnvironment:
+    def create_plugin(self, trainer_config):
         from lightning.fabric.plugins.environments.torchelastic import (
             TorchElasticEnvironment,
         )
@@ -117,12 +175,12 @@ class TorchElasticEnvironmentPlugin(PluginConfigBase):
 
 @plugin_registry.register
 class XLAEnvironmentPlugin(PluginConfigBase):
-    name: Literal["xla_environment"] = "xla_environment"
-
     """Cluster environment for training on a TPU Pod with the PyTorch/XLA library."""
 
+    name: Literal["xla_environment"] = "xla_environment"
+
     @override
-    def create_plugin(self, trainer_config) -> ClusterEnvironment:
+    def create_plugin(self, trainer_config):
         from lightning.fabric.plugins.environments.xla import XLAEnvironment
 
         return XLAEnvironment()

nshtrainer/trainer/plugin/io.py

@@ -2,27 +2,52 @@ from __future__ import annotations
 
 from typing import Literal
 
-from lightning.pytorch.plugins.io import CheckpointIO
 from typing_extensions import override
 
-from .base import PluginConfig, PluginConfigBase, plugin_registry
+from .base import PluginConfigBase, plugin_registry
 
 
 @plugin_registry.register
-class AsyncCheckpointIOPlugin(PluginConfigBase):
-    name: Literal["async_checkpoint"] = "async_checkpoint"
+class TorchCheckpointIOPlugin(PluginConfigBase):
+    """CheckpointIO that utilizes torch.save and torch.load to save and load checkpoints respectively."""
+
+    name: Literal["torch_checkpoint"] = "torch_checkpoint"
+
+    @override
+    def create_plugin(self, trainer_config):
+        from lightning.fabric.plugins.io.torch_io import TorchCheckpointIO
+
+        return TorchCheckpointIO()
+
+
+@plugin_registry.register
+class XLACheckpointIOPlugin(PluginConfigBase):
+    """CheckpointIO that utilizes xm.save to save checkpoints for TPU training strategies."""
 
+    name: Literal["xla_checkpoint"] = "xla_checkpoint"
+
+    @override
+    def create_plugin(self, trainer_config):
+        from lightning.fabric.plugins.io.xla import XLACheckpointIO
+
+        return XLACheckpointIO()
+
+
+@plugin_registry.register
+class AsyncCheckpointIOPlugin(PluginConfigBase):
     """Enables saving the checkpoints asynchronously in a thread.
 
     .. warning::  This is an experimental feature.
     """
 
-    checkpoint_io: PluginConfig | None = None
+    name: Literal["async_checkpoint"] = "async_checkpoint"
+
+    checkpoint_io: TorchCheckpointIOPlugin | None = None
     """A checkpoint IO plugin that is used as the basis for async checkpointing."""
 
     @override
-    def create_plugin(self, trainer_config) -> CheckpointIO:
-        from lightning.pytorch.plugins.io.async_plugin import AsyncCheckpointIO
+    def create_plugin(self, trainer_config):
+        from lightning.pytorch.plugins.io import AsyncCheckpointIO, CheckpointIO
 
         base_io = (
             self.checkpoint_io.create_plugin(trainer_config)
@@ -34,29 +59,3 @@ class AsyncCheckpointIOPlugin(PluginConfigBase):
                 f"Expected `checkpoint_io` to be a `CheckpointIO` instance, but got {type(base_io)}."
             )
         return AsyncCheckpointIO(checkpoint_io=base_io)
-
-
-@plugin_registry.register
-class TorchCheckpointIOPlugin(PluginConfigBase):
-    name: Literal["torch_checkpoint"] = "torch_checkpoint"
-
-    """CheckpointIO that utilizes torch.save and torch.load to save and load checkpoints respectively."""
-
-    @override
-    def create_plugin(self, trainer_config) -> CheckpointIO:
-        from lightning.fabric.plugins.io.torch_io import TorchCheckpointIO
-
-        return TorchCheckpointIO()
-
-
-@plugin_registry.register
-class XLACheckpointIOPlugin(PluginConfigBase):
-    name: Literal["xla_checkpoint"] = "xla_checkpoint"
-
-    """CheckpointIO that utilizes xm.save to save checkpoints for TPU training strategies."""
-
-    @override
-    def create_plugin(self, trainer_config) -> CheckpointIO:
-        from lightning.fabric.plugins.io.xla import XLACheckpointIO
-
-        return XLACheckpointIO()

nshtrainer/trainer/plugin/layer_sync.py

@@ -2,7 +2,6 @@ from __future__ import annotations
 
 from typing import Literal
 
-from lightning.pytorch.plugins.layer_sync import LayerSync
 from typing_extensions import override
 
 from .base import PluginConfigBase, plugin_registry
@@ -10,16 +9,16 @@ from .base import PluginConfigBase, plugin_registry
 
 @plugin_registry.register
 class TorchSyncBatchNormPlugin(PluginConfigBase):
-    name: Literal["torch_sync_batchnorm"] = "torch_sync_batchnorm"
-
     """A plugin that wraps all batch normalization layers of a model with synchronization
     logic for multiprocessing.
 
     This plugin has no effect in single-device operation.
     """
 
+    name: Literal["torch_sync_batchnorm"] = "torch_sync_batchnorm"
+
     @override
-    def create_plugin(self, trainer_config) -> LayerSync:
+    def create_plugin(self, trainer_config):
         from lightning.pytorch.plugins.layer_sync import TorchSyncBatchNorm
 
         return TorchSyncBatchNorm()

nshtrainer/trainer/plugin/precision.py

@@ -2,7 +2,6 @@ from __future__ import annotations
 
 from typing import Any, Literal
 
-from lightning.pytorch.plugins.precision import Precision
 from typing_extensions import override
 
 from ...util.config.dtype import DTypeConfig
@@ -20,7 +19,7 @@ class MixedPrecisionPluginConfig(PluginConfigBase):
     """The device for ``torch.autocast``."""
 
     @override
-    def create_plugin(self, trainer_config) -> Precision:
+    def create_plugin(self, trainer_config):
         from lightning.pytorch.plugins.precision.amp import MixedPrecision
 
         return MixedPrecision(self.precision, self.device)
@@ -45,7 +44,7 @@ class BitsandbytesPluginConfig(PluginConfigBase):
     """
 
     @override
-    def create_plugin(self, trainer_config) -> Precision:
+    def create_plugin(self, trainer_config):
         from lightning.pytorch.plugins.precision.bitsandbytes import (
             BitsandbytesPrecision,
         )
@@ -66,7 +65,7 @@ class DeepSpeedPluginConfig(PluginConfigBase):
     mixed precision (16-mixed, bf16-mixed)."""
 
     @override
-    def create_plugin(self, trainer_config) -> Precision:
+    def create_plugin(self, trainer_config):
         from lightning.pytorch.plugins.precision.deepspeed import DeepSpeedPrecision
 
         return DeepSpeedPrecision(precision=self.precision)
@@ -80,7 +79,7 @@ class DoublePrecisionPluginConfig(PluginConfigBase):
     """Plugin for training with double (``torch.float64``) precision."""
 
     @override
-    def create_plugin(self, trainer_config) -> Precision:
+    def create_plugin(self, trainer_config):
         from lightning.pytorch.plugins.precision.double import DoublePrecision
 
         return DoublePrecision()
@@ -95,7 +94,7 @@ class FSDPPrecisionPluginConfig(PluginConfigBase):
     mixed precision (16-mixed, bf16-mixed)."""
 
     @override
-    def create_plugin(self, trainer_config) -> Precision:
+    def create_plugin(self, trainer_config):
         from lightning.pytorch.plugins.precision.fsdp import FSDPPrecision
 
         return FSDPPrecision(precision=self.precision)
@@ -109,7 +108,7 @@ class HalfPrecisionPluginConfig(PluginConfigBase):
     """Whether to use ``torch.float16`` (``'16-true'``) or ``torch.bfloat16`` (``'bf16-true'``)."""
 
     @override
-    def create_plugin(self, trainer_config) -> Precision:
+    def create_plugin(self, trainer_config):
         from lightning.pytorch.plugins.precision.half import HalfPrecision
 
         return HalfPrecision(precision=self.precision)
@@ -134,7 +133,7 @@ class TransformerEnginePluginConfig(PluginConfigBase):
     Defaults to the same as weights_dtype."""
 
     @override
-    def create_plugin(self, trainer_config) -> Precision:
+    def create_plugin(self, trainer_config):
         from lightning.pytorch.plugins.precision.transformer_engine import (
             TransformerEnginePrecision,
         )
@@ -157,7 +156,7 @@ class XLAPluginConfig(PluginConfigBase):
     """Full precision (32-true) or half precision (16-true, bf16-true)."""
 
     @override
-    def create_plugin(self, trainer_config) -> Precision:
+    def create_plugin(self, trainer_config):
         from lightning.pytorch.plugins.precision.xla import XLAPrecision
 
         return XLAPrecision(precision=self.precision)

{nshtrainer-1.0.0b55.dist-info → nshtrainer-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: nshtrainer
-Version: 1.0.0b55
+Version: 1.1.0
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com
@@ -14,13 +14,12 @@ Provides-Extra: extra
 Requires-Dist: GitPython ; extra == "extra"
 Requires-Dist: huggingface-hub ; extra == "extra"
 Requires-Dist: lightning
-Requires-Dist: nshconfig
+Requires-Dist: nshconfig (>0.39)
 Requires-Dist: nshrunner
 Requires-Dist: nshutils ; extra == "extra"
 Requires-Dist: numpy
 Requires-Dist: packaging
 Requires-Dist: psutil
-Requires-Dist: pytorch-lightning
 Requires-Dist: tensorboard ; extra == "extra"
 Requires-Dist: torch
 Requires-Dist: torchmetrics

{nshtrainer-1.0.0b55.dist-info → nshtrainer-1.1.0.dist-info}/RECORD

@@ -87,9 +87,9 @@ nshtrainer/configs/trainer/__init__.py,sha256=a8pzGVid52abAVARPbgjaN566H1ZM44FH_
 nshtrainer/configs/trainer/_config/__init__.py,sha256=6DXdtP-uH11TopQ7kzId9fco-wVkD7ZfevbBqDpN6TE,3817
 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
-nshtrainer/configs/trainer/plugin/environment/__init__.py,sha256=3o16x4qRAOvkJH9Vg4-QwsEODDC6aP_OXRnPPkm_xSo,1376
-nshtrainer/configs/trainer/plugin/io/__init__.py,sha256=W6G67JnigB6d3MiwLrbSKgtIZLUccXznp-IXwkK1J4U,743
+nshtrainer/configs/trainer/plugin/base/__init__.py,sha256=slW5z1FZw2qICXO9l9DnLIDB1Yl7KOcxPEZkyYIHrp4,276
+nshtrainer/configs/trainer/plugin/environment/__init__.py,sha256=geTaOfbFep3DVDgSsRrufM7q6MIZur5QG1_47ngAL0I,1280
+nshtrainer/configs/trainer/plugin/io/__init__.py,sha256=AtGUuE0M16dTpX0q9NqvJiE4qU1j07N0RLtd-JFzWuc,653
 nshtrainer/configs/trainer/plugin/layer_sync/__init__.py,sha256=SYDZk2M6sgpt4sEuoURuS8EKYmaqGcvYxETE9jvTrEE,431
 nshtrainer/configs/trainer/plugin/precision/__init__.py,sha256=szlqSfK2XuWdkf72LQzQFv3SlWfKFdRUpBEYIxQ3TPs,1507
 nshtrainer/configs/trainer/strategy/__init__.py,sha256=50whNloJVBq_bdbLaPQnPBTeS1Rcs8MwxTCYBj1kKa4,273
@@ -136,12 +136,12 @@ nshtrainer/trainer/__init__.py,sha256=fQ7gQRlGWX-90TYT0rttkQyvXDCzo7DAvJgr-jX1zs
 nshtrainer/trainer/_config.py,sha256=s-_XoLc9mbNAdroRJyOKd3dLTyrFLQkPyGJkKDmBYf8,33267
 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
-nshtrainer/trainer/plugin/base.py,sha256=9-qUHXGpll_yCylun0899sbmJDpyhD9IQcBtVrJx38I,919
-nshtrainer/trainer/plugin/environment.py,sha256=NW0qbsbvDPe59JGOMgPLq1fj7szLucIV1WRTxCrcjF4,4367
-nshtrainer/trainer/plugin/io.py,sha256=nm6YDCVZAhmPvLaLnw6q4BrK2Gj2wvD5ZLDhj1xneEE,2030
-nshtrainer/trainer/plugin/layer_sync.py,sha256=h-ydZwXepnsw5-paLgiDatqPyQ_8C0QEv6DrYFIaXOo,735
-nshtrainer/trainer/plugin/precision.py,sha256=I0QsB1bVxmsFmBOkgrAfGONsuYae_lD9Bz0PfJEQvH4,5598
+nshtrainer/trainer/plugin/__init__.py,sha256=q_q98MYNaZ2VE_tqGqYlQjQnlaF4NE1FUqVVbj0EK7k,517
+nshtrainer/trainer/plugin/base.py,sha256=76ct2TTHLpPr5MO8B9CIkoCOo-dFImzqAll8cIdC0cg,736
+nshtrainer/trainer/plugin/environment.py,sha256=SSXRWHjyFUA6oFx3duD_ZwhM59pWUjR1_UzHz02NI2c,5440
+nshtrainer/trainer/plugin/io.py,sha256=OmFSKLloMypletjaUr_Ptg6LS0ljqTVIp2o4Hm3eZoE,1926
+nshtrainer/trainer/plugin/layer_sync.py,sha256=-BbEyWZ063O7tZme7Gdu1lVxK6p1NeuLcOfPXnvH29s,663
+nshtrainer/trainer/plugin/precision.py,sha256=7lf7KZd_yFyPmhLApjEIv0pkoDB5zdxi-7in0wRj3z8,5436
 nshtrainer/trainer/signal_connector.py,sha256=GhfGcSzfaTNhnj2QFkBDq5aT7FqbLMA7eC8SYQs8_8w,10828
 nshtrainer/trainer/strategy.py,sha256=VPTn5z3zvXTydY8IJchjhjcOfpvtoejnvUkq5E4WTus,1368
 nshtrainer/trainer/trainer.py,sha256=Lo3vUo3ooTAjaX2fUYPFSMv5FP7sWfVov0QbA-T5hZ8,21113
@@ -156,6 +156,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.0b55.dist-info/METADATA,sha256=3JHfGqw8kR8FKZT-E--W3H_rS5M-QQHH9U2EooKyO70,988
-nshtrainer-1.0.0b55.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-nshtrainer-1.0.0b55.dist-info/RECORD,,
+nshtrainer-1.1.0.dist-info/METADATA,sha256=9prl1vhkmagIgO0WmzfDEknzv4S0a0qDAh-Bj3LxFjI,960
+nshtrainer-1.1.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+nshtrainer-1.1.0.dist-info/RECORD,,