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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (4) hide show
  1. nshtrainer/model/config.py +10 -0
  2. {nshtrainer-0.40.1.dist-info → nshtrainer-0.40.2.dist-info}/METADATA +1 -1
  3. {nshtrainer-0.40.1.dist-info → nshtrainer-0.40.2.dist-info}/RECORD +4 -4
  4. {nshtrainer-0.40.1.dist-info → nshtrainer-0.40.2.dist-info}/WHEEL +0 -0
nshtrainer/model/config.py CHANGED
@@ -77,6 +77,16 @@ class BaseConfig(C.Config):
77
77
  return self.directory.resolve_subdirectory(self.id, subdirectory)
78
78
 
79
79
  # region Helper methods
80
+ def fast_dev_run(self, value: int | bool, /):
81
+ """
82
+ Enables fast_dev_run mode for the trainer.
83
+ This will run the training loop for a specified number of batches,
84
+ if an integer is provided, or for a single batch if True is provided.
85
+ """
86
+ config = copy.deepcopy(self)
87
+ config.trainer.fast_dev_run = value
88
+ return config
89
+
80
90
  def with_project_root_(self, project_root: str | Path | os.PathLike) -> Self:
81
91
  """
82
92
  Set the project root directory for the trainer.
{nshtrainer-0.40.1.dist-info → nshtrainer-0.40.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.40.1
3
+ Version: 0.40.2
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.40.1.dist-info → nshtrainer-0.40.2.dist-info}/RECORD RENAMED
@@ -66,7 +66,7 @@ nshtrainer/metrics/__init__.py,sha256=ObLIELGguIEcUpRsUkqh1ltrvZii6vglTpJGrPvoy0
66
66
  nshtrainer/metrics/_config.py,sha256=jgRBfDAQLFTW7AiUY7CRtdfts6CR6keeuqm0FFMWCzQ,1288
67
67
  nshtrainer/model/__init__.py,sha256=2i_VEy6u_Y1LUGKljHXWeekvhnUcanZM2QyaaBM1Bmw,261
68
68
  nshtrainer/model/base.py,sha256=NasbYZJBuEly6Hm9t9HVZk-CUHmy4T7p1v-Ye981XA4,18609
69
- nshtrainer/model/config.py,sha256=Q4Wong6w3cp_Sq7s8iZdABKF-LZBbSCFn_TQPYkhkrI,6572
69
+ nshtrainer/model/config.py,sha256=tZg6qsHwyhYFLe7MQx_y2HRPVgnt8WZAB3DpsOb0uso,6954
70
70
  nshtrainer/model/mixins/callback.py,sha256=rbe8P22iEjPkH1df6rfEo3Txw7EwSz6Dkm0TWO_AysM,2419
71
71
  nshtrainer/model/mixins/logger.py,sha256=xOymSTofukEYZGkGojXsMEO__ZlBI5lIPZVmlotMEX8,5291
72
72
  nshtrainer/nn/__init__.py,sha256=0QPFl02a71WZQjLMGOlFNMmsYP5aa1q3eABHmnWH58Q,1427
@@ -100,6 +100,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
100
100
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
101
101
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
102
102
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
103
- nshtrainer-0.40.1.dist-info/METADATA,sha256=m9hT1zAMnexEhlbXEvTFJBbDsWffVWw11M9NxqqTrZg,916
104
- nshtrainer-0.40.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
105
- nshtrainer-0.40.1.dist-info/RECORD,,
103
+ nshtrainer-0.40.2.dist-info/METADATA,sha256=yeuMHfYVmcPkNcPveUU1Rq38HeIunxERh4gqvzy85Uc,916
104
+ nshtrainer-0.40.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
105
+ nshtrainer-0.40.2.dist-info/RECORD,,