nshtrainer 0.10.5__py3-none-any.whl → 0.10.6__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 (5) hide show
  1. nshtrainer/model/config.py +3 -0
  2. nshtrainer/trainer/trainer.py +4 -0
  3. {nshtrainer-0.10.5.dist-info → nshtrainer-0.10.6.dist-info}/METADATA +1 -1
  4. {nshtrainer-0.10.5.dist-info → nshtrainer-0.10.6.dist-info}/RECORD +5 -5
  5. {nshtrainer-0.10.5.dist-info → nshtrainer-0.10.6.dist-info}/WHEEL +0 -0
nshtrainer/model/config.py CHANGED
@@ -1121,6 +1121,9 @@ class SanityCheckingConfig(C.Config):
1121
1121
 
1122
1122
 
1123
1123
  class TrainerConfig(C.Config):
1124
+ ckpt_path: str | Path | None = None
1125
+ """Path to a checkpoint to load and resume training from."""
1126
+
1124
1127
  checkpoint_loading: CheckpointLoadingConfig | Literal["auto"] = "auto"
1125
1128
  """Checkpoint loading configuration options."""
1126
1129
 
nshtrainer/trainer/trainer.py CHANGED
@@ -304,6 +304,10 @@ class Trainer(LightningTrainer):
304
304
  log_dir = str(Path(log_dir).resolve())
305
305
  log.critical(f"LightningTrainer log directory: {self.log_dir}.")
306
306
 
307
+ # Set the checkpoint
308
+ if (ckpt_path := config.trainer.ckpt_path) is not None:
309
+ self.ckpt_path = str(Path(ckpt_path).resolve().absolute())
310
+
307
311
  def __runtime_tracker(self):
308
312
  return next(
309
313
  (
{nshtrainer-0.10.5.dist-info → nshtrainer-0.10.6.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.10.5
3
+ Version: 0.10.6
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.10.5.dist-info → nshtrainer-0.10.6.dist-info}/RECORD RENAMED
@@ -52,7 +52,7 @@ nshtrainer/metrics/_config.py,sha256=hWWS4IXENRyH3RmJ7z1Wx1n3Lt1sNMlGOrcU6PW15o0
52
52
  nshtrainer/model/__init__.py,sha256=TbexTxiE20WHYg5q3L88Hysk4LlHeKk_isv33aSBREA,1918
53
53
  nshtrainer/model/_environment.py,sha256=JCFxxwMhkviiMDkqIXJmiuepqiSYIlcoSQM7Y2H2KX4,23036
54
54
  nshtrainer/model/base.py,sha256=Bmw-t70TydDbE9P0ee-lTibGoUhrCx5Qke-upa7FGVM,17512
55
- nshtrainer/model/config.py,sha256=OsVba02cmEYVf6V-A6ljV7VMAW5XZO6GWNRk8ktUw2o,53177
55
+ nshtrainer/model/config.py,sha256=6pAqDUk1eBloR3vZmtsWVdMrKeT2V3UvOn5UZ7YhZ_Q,53283
56
56
  nshtrainer/model/modules/callback.py,sha256=JF59U9-CjJsAIspEhTJbVaGN0wGctZG7UquE3IS7R8A,6408
57
57
  nshtrainer/model/modules/debug.py,sha256=DTVty8cKnzj1GCULRyGx_sWTTsq9NLi30dzqjRTnuCU,1127
58
58
  nshtrainer/model/modules/distributed.py,sha256=ABpR9d-3uBS_fivfy_WYW-dExW6vp5BPaoPQnOudHng,1725
@@ -73,12 +73,12 @@ nshtrainer/trainer/__init__.py,sha256=P2rmr8oBVTHk-HJHYPcUwWqDEArMbPR4_rPpATbWK3
73
73
  nshtrainer/trainer/_runtime_callback.py,sha256=sd2cUdRJG-UCdQr9ruZvEYpNGNF1t2W2fuxwwVlQD9E,4164
74
74
  nshtrainer/trainer/checkpoint_connector.py,sha256=xoqI2dcPnlNFPPLVIU6dBOvRPC9PtfX5qu__xV1lx0Y,2124
75
75
  nshtrainer/trainer/signal_connector.py,sha256=llwc8pdKAWxREFpjdi14Bpy8rGVMEJsmJx_s2p4gI8E,10689
76
- nshtrainer/trainer/trainer.py,sha256=n3T9Iz3eaDostxEdjapWImAsVMxyU9WBdhlPl0THX-g,16785
76
+ nshtrainer/trainer/trainer.py,sha256=tFyzIsF8c-FABTH6wwDOR9y8kydVJqeVO7PDNFMvhSU,16950
77
77
  nshtrainer/util/environment.py,sha256=_SEtiQ_s5bL5pllUlf96AOUv15kNvCPvocVC13S7mIk,4166
78
78
  nshtrainer/util/seed.py,sha256=HEXgVs-wldByahOysKwq7506OHxdYTEgmP-tDQVAEkQ,287
79
79
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
80
80
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
81
81
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
82
- nshtrainer-0.10.5.dist-info/METADATA,sha256=nBa8n5rSpkY6MWYI-2JjmJzWKWa7gY0NOs737jsXXsU,695
83
- nshtrainer-0.10.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
84
- nshtrainer-0.10.5.dist-info/RECORD,,
82
+ nshtrainer-0.10.6.dist-info/METADATA,sha256=YXr1jz_lfmXzwsNBxyYtSh6CbqAJ_wcGdAzkmZoCjH4,695
83
+ nshtrainer-0.10.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
84
+ nshtrainer-0.10.6.dist-info/RECORD,,