nshtrainer 0.40.0__py3-none-any.whl → 0.40.2__py3-none-any.whl

nshtrainer/loggers/wandb.py

@@ -92,7 +92,7 @@ class WandbLoggerConfig(CallbackConfigBase, BaseLoggerConfig):
     - "none" or False: Do not log any checkpoints
     """
 
-    log_code: WandbUploadCodeConfig | None = None
+    log_code: WandbUploadCodeConfig | None = WandbUploadCodeConfig()
     """WandB code upload configuration. Used to upload code to WandB."""
 
     watch: WandbWatchConfig | None = WandbWatchConfig()

nshtrainer/model/config.py

@@ -77,6 +77,16 @@ class BaseConfig(C.Config):
         return self.directory.resolve_subdirectory(self.id, subdirectory)
 
     # region Helper methods
+    def fast_dev_run(self, value: int | bool, /):
+        """
+        Enables fast_dev_run mode for the trainer.
+        This will run the training loop for a specified number of batches,
+        if an integer is provided, or for a single batch if True is provided.
+        """
+        config = copy.deepcopy(self)
+        config.trainer.fast_dev_run = value
+        return config
+
     def with_project_root_(self, project_root: str | Path | os.PathLike) -> Self:
         """
         Set the project root directory for the trainer.

{nshtrainer-0.40.0.dist-info → nshtrainer-0.40.2.dist-info}/METADATA

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

{nshtrainer-0.40.0.dist-info → nshtrainer-0.40.2.dist-info}/RECORD

@@ -57,7 +57,7 @@ nshtrainer/loggers/__init__.py,sha256=C_xk0A3_qKbNdTmzK85AgjRHFD3w-jPRS2ig-iPhfE
 nshtrainer/loggers/_base.py,sha256=xiZKEK0ALJkcqf4OpVNRY0QbZsamR_WR7x7m_68YHXQ,705
 nshtrainer/loggers/csv.py,sha256=D_lYyd94bZ8jAgnRo-ARtFgVcInaD9zktxtsUD9RWCI,1052
 nshtrainer/loggers/tensorboard.py,sha256=wL2amRSdP68zbslZvBeM0ZQBnjF3hIKsz-_lBbdomaM,2216
-nshtrainer/loggers/wandb.py,sha256=td8J2v8T1nvGQI7OYQ1El6k8FGsXZxbnuY97s8KzCiY,6643
+nshtrainer/loggers/wandb.py,sha256=GMlEzkFesgS3qgVgYqATPHJ3rvs9etwFlk3HbidOyRQ,6662
 nshtrainer/lr_scheduler/__init__.py,sha256=uEvgaFAs-4s_bAEMaildy0GT6OvgpgOEKTuzqutESHE,736
 nshtrainer/lr_scheduler/_base.py,sha256=7xOIuxQ86YHbFWG5a3gX46emQj1WN_LaY4-i0Q1TDBg,3659
 nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=E8LW78uuby7bIsoLPpcF1bmNK4lSko-r3qPL-vuHWXQ,5370
@@ -66,7 +66,7 @@ nshtrainer/metrics/__init__.py,sha256=ObLIELGguIEcUpRsUkqh1ltrvZii6vglTpJGrPvoy0
 nshtrainer/metrics/_config.py,sha256=jgRBfDAQLFTW7AiUY7CRtdfts6CR6keeuqm0FFMWCzQ,1288
 nshtrainer/model/__init__.py,sha256=2i_VEy6u_Y1LUGKljHXWeekvhnUcanZM2QyaaBM1Bmw,261
 nshtrainer/model/base.py,sha256=NasbYZJBuEly6Hm9t9HVZk-CUHmy4T7p1v-Ye981XA4,18609
-nshtrainer/model/config.py,sha256=Q4Wong6w3cp_Sq7s8iZdABKF-LZBbSCFn_TQPYkhkrI,6572
+nshtrainer/model/config.py,sha256=tZg6qsHwyhYFLe7MQx_y2HRPVgnt8WZAB3DpsOb0uso,6954
 nshtrainer/model/mixins/callback.py,sha256=rbe8P22iEjPkH1df6rfEo3Txw7EwSz6Dkm0TWO_AysM,2419
 nshtrainer/model/mixins/logger.py,sha256=xOymSTofukEYZGkGojXsMEO__ZlBI5lIPZVmlotMEX8,5291
 nshtrainer/nn/__init__.py,sha256=0QPFl02a71WZQjLMGOlFNMmsYP5aa1q3eABHmnWH58Q,1427
@@ -100,6 +100,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
 nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
 nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
 nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
-nshtrainer-0.40.0.dist-info/METADATA,sha256=E8dGtOSZ6jdCxugNpIutAeUyd7CcYSRH9cBchBDY7UE,916
-nshtrainer-0.40.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-0.40.0.dist-info/RECORD,,
+nshtrainer-0.40.2.dist-info/METADATA,sha256=yeuMHfYVmcPkNcPveUU1Rq38HeIunxERh4gqvzy85Uc,916
+nshtrainer-0.40.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-0.40.2.dist-info/RECORD,,