nshtrainer 1.0.0b18__py3-none-any.whl → 1.0.0b20__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
@@ -824,6 +822,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/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"):
@@ -245,6 +249,8 @@ class Trainer(LightningTrainer):
                 if logger_config is not None
                 and (logger := logger_config.create_logger(hparams)) is not None
             ]
+            if not hparams.barebones
+            else None
         )
 
         if hparams.auto_determine_num_nodes:

{nshtrainer-1.0.0b18.dist-info → nshtrainer-1.0.0b20.dist-info}/METADATA

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

{nshtrainer-1.0.0b18.dist-info → nshtrainer-1.0.0b20.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=ymHI-XdNsYPr-jGUFWFh4scuIN0dHWa5od6-kyBpvYQ,33850
+nshtrainer/trainer/_config.py,sha256=I6DBPnkveBTZOL01a8be5VK1YL-k6h3ubEwJi7FxKlk,33912
 nshtrainer/trainer/_runtime_callback.py,sha256=T3epaj1YeIN0R8CS2cg5HNJIB21TyaD_PVNNOPJ6nJs,4200
 nshtrainer/trainer/signal_connector.py,sha256=YMJf6vTnW0JcnBkuYikm9x_9XscaokrCEzCn4THOGao,10776
-nshtrainer/trainer/trainer.py,sha256=78P9b4PYL7ez48U_0N4AgUAeTb_UQHXIU8EwPRY7b1w,19431
+nshtrainer/trainer/trainer.py,sha256=LYOjvonQ6_F9thFuOpVKRBUf2V6gQltW8NmbwkiNbXA,19590
 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.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,,
+nshtrainer-1.0.0b20.dist-info/METADATA,sha256=Sk2yjQENY7vB9ZzMtUTXJAcPEOmlDX0RaaGQUSQHn28,937
+nshtrainer-1.0.0b20.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-1.0.0b20.dist-info/RECORD,,