nshtrainer 1.0.0b17__py3-none-any.whl → 1.0.0b19__py3-none-any.whl

nshtrainer/trainer/_config.py

@@ -732,9 +732,7 @@ class TrainerConfig(C.Config):
     automatic selection based on the chosen accelerator. Default: ``"auto"``.
     """
 
-    shared_parameters: SharedParametersCallbackConfig | None = (
-        SharedParametersCallbackConfig()
-    )
+    shared_parameters: SharedParametersCallbackConfig | None = None
     """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
@@ -802,6 +800,10 @@ class TrainerConfig(C.Config):
         )
 
     def _nshtrainer_all_callback_configs(self) -> Iterable[CallbackConfigBase | None]:
+        # Disable all callbacks if barebones mode is enabled
+        if self.barebones:
+            return
+
         yield self.early_stopping
         yield self.checkpoint_saving
         yield self.lr_monitor
@@ -823,6 +825,11 @@ class TrainerConfig(C.Config):
         yield from self.enabled_loggers()
         yield self.actsave_logger
 
+    def _nshtrainer_validate_before_run(self):
+        # shared_parameters is not supported under barebones mode
+        if self.barebones and self.shared_parameters:
+            raise ValueError("shared_parameters is not supported under barebones mode")
+
     # region Helper Methods
     def fast_dev_run_(self, value: int | bool = True, /):
         """

nshtrainer/trainer/trainer.py

@@ -134,6 +134,10 @@ class Trainer(LightningTrainer):
             for key, value in update.items():
                 _update_key(key, value)
 
+        # Set `barebones`
+        if hparams.barebones:
+            _update_kwargs(barebones=True)
+
         # Set `default_root_dir` if `auto_set_default_root_dir` is enabled.
         if hparams.auto_set_default_root_dir:
             if kwargs.get("default_root_dir"):
@@ -296,6 +300,7 @@ class Trainer(LightningTrainer):
                 f"Got {type(hparams)=} instead."
             )
         hparams = hparams.model_deep_validate()
+        hparams._nshtrainer_validate_before_run()
 
         self._pre_init(hparams)
 

{nshtrainer-1.0.0b17.dist-info → nshtrainer-1.0.0b19.dist-info}/METADATA

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

{nshtrainer-1.0.0b17.dist-info → nshtrainer-1.0.0b19.dist-info}/RECORD

@@ -119,10 +119,10 @@ nshtrainer/profiler/advanced.py,sha256=XrM3FX0ThCv5UwUrrH0l4Ow4LGAtpiBww2N8QAU5N
 nshtrainer/profiler/pytorch.py,sha256=8K37XvPnCApUpIK8tA2zNMFIaIiTLSoxKQoiyCPBm1Q,2757
 nshtrainer/profiler/simple.py,sha256=PimjqcU-JuS-8C0ZGHAdwCxgNLij4x0FH6WXsjBQzZs,1005
 nshtrainer/trainer/__init__.py,sha256=MmoydVS6aYeav7zgDAUHxAQrV_PMQsbnZTCuPnLH9Wk,128
-nshtrainer/trainer/_config.py,sha256=Bx-32uNvmJ68qlRoK8xxh71gZcqkIkDJNkkOWoU5Hnc,33486
+nshtrainer/trainer/_config.py,sha256=2AIr8w_ysRtn0yo49rwdduyBJ9bIAVpQdRpJoMg9Cd0,33806
 nshtrainer/trainer/_runtime_callback.py,sha256=T3epaj1YeIN0R8CS2cg5HNJIB21TyaD_PVNNOPJ6nJs,4200
 nshtrainer/trainer/signal_connector.py,sha256=YMJf6vTnW0JcnBkuYikm9x_9XscaokrCEzCn4THOGao,10776
-nshtrainer/trainer/trainer.py,sha256=vfQAr5H5HkDlIxjdEP8yhLDKplodxIws3sx3u_8qbkc,19381
+nshtrainer/trainer/trainer.py,sha256=V5aRA6hBSxYi-Hbp-lg6b5mRCw_bc_0QzkJ7LG0c49M,19531
 nshtrainer/util/_environment_info.py,sha256=MT8mBe6ZolRfKiwU-les1P-lPNPqXpHQcfADrh_A3uY,24629
 nshtrainer/util/_useful_types.py,sha256=dwZokFkIe7M5i2GR3nQ9A1lhGw06DMAFfH5atyquqSA,8000
 nshtrainer/util/bf16.py,sha256=9QhHZCkYSfYpIcxwAMoXyuh2yTSHBzT-EdLQB297jEs,762
@@ -135,6 +135,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.0b17.dist-info/METADATA,sha256=6-ZUgdzm04noloz7OD0Y3jxg-pBe2u-WqYS4ZGFhWIU,937
-nshtrainer-1.0.0b17.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-1.0.0b17.dist-info/RECORD,,
+nshtrainer-1.0.0b19.dist-info/METADATA,sha256=5WTNG0hJXpVLs3iZrpsIjgbccJeuIMG7i7t4MyAeIt4,937
+nshtrainer-1.0.0b19.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-1.0.0b19.dist-info/RECORD,,