nshtrainer 1.0.0b19__tar.gz → 1.0.0b21__tar.gz

{nshtrainer-1.0.0b19 → nshtrainer-1.0.0b21}/PKG-INFO

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

{nshtrainer-1.0.0b19 → nshtrainer-1.0.0b21}/pyproject.toml

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

{nshtrainer-1.0.0b19 → nshtrainer-1.0.0b21}/src/nshtrainer/trainer/_config.py

@@ -602,6 +602,10 @@ class TrainerConfig(C.Config):
     :meth:`~lightning.pytorch.core.LightningModule.log_dict`.
     """
 
+    experimental_barebones_profiler_enabled: bool = False
+    """If enabled, will enable the profiler in barebones mode.
+    The implementation of this is very hacky and may not work as expected."""
+
     precision: (
         Literal[
             "64-true",
@@ -822,6 +826,10 @@ class TrainerConfig(C.Config):
         yield from self.callbacks
 
     def _nshtrainer_all_logger_configs(self) -> Iterable[BaseLoggerConfig | None]:
+        # Disable all loggers if barebones mode is enabled
+        if self.barebones:
+            return
+
         yield from self.enabled_loggers()
         yield self.actsave_logger
 

{nshtrainer-1.0.0b19 → nshtrainer-1.0.0b21}/src/nshtrainer/trainer/trainer.py

@@ -278,6 +278,12 @@ class Trainer(LightningTrainer):
         _update_kwargs(**hparams.lightning_kwargs)
         _update_kwargs(**kwargs_ctor)
 
+        # Handle barebones mode
+        if kwargs.get("barebones"):
+            # Remove the logger if it's an empty list
+            if (logger := kwargs.get("logger")) is not None and not logger:
+                kwargs["logger"] = None
+
         return kwargs
 
     if TYPE_CHECKING: