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

nshtrainer/trainer/_config.py

@@ -802,6 +802,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 +827,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

@@ -296,6 +296,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.0b18.dist-info}/METADATA

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

{nshtrainer-1.0.0b17.dist-info → nshtrainer-1.0.0b18.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=ymHI-XdNsYPr-jGUFWFh4scuIN0dHWa5od6-kyBpvYQ,33850
 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=78P9b4PYL7ez48U_0N4AgUAeTb_UQHXIU8EwPRY7b1w,19431
 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.0b18.dist-info/METADATA,sha256=MMBVxbqYv4H-wrmkb5xmxINe_7yWTJFAtfBJoew_tDM,937
+nshtrainer-1.0.0b18.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-1.0.0b18.dist-info/RECORD,,