nshtrainer 1.0.0b16__tar.gz → 1.0.0b18__tar.gz

{nshtrainer-1.0.0b16 → nshtrainer-1.0.0b18}/PKG-INFO

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

{nshtrainer-1.0.0b16 → nshtrainer-1.0.0b18}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshtrainer"
-version = "1.0.0-beta16"
+version = "1.0.0-beta18"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"

{nshtrainer-1.0.0b16 → nshtrainer-1.0.0b18}/src/nshtrainer/trainer/_config.py

@@ -585,7 +585,7 @@ class TrainerConfig(C.Config):
     Default: ``False``.
     """
 
-    barebones: bool
+    barebones: bool = False
     """Whether to run in "barebones mode", where all features that may impact raw speed are
     disabled. This is meant for analyzing the Trainer overhead and is discouraged during regular training
     runs. The following features are deactivated:
@@ -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-1.0.0b16 → nshtrainer-1.0.0b18}/src/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)