nshtrainer 0.11.4__py3-none-any.whl → 0.11.5__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/callbacks/checkpoint/best_checkpoint.py +4 -3
  2. {nshtrainer-0.11.4.dist-info → nshtrainer-0.11.5.dist-info}/METADATA +1 -1
  3. {nshtrainer-0.11.4.dist-info → nshtrainer-0.11.5.dist-info}/RECORD +4 -4
  4. {nshtrainer-0.11.4.dist-info → nshtrainer-0.11.5.dist-info}/WHEEL +0 -0
nshtrainer/callbacks/checkpoint/best_checkpoint.py CHANGED
@@ -150,6 +150,7 @@ class BestCheckpoint(Checkpoint):
150
150
  # Save the current model
151
151
  filepath = self._ckpt_path(trainer)
152
152
  trainer.save_checkpoint(filepath, self.config.save_weights_only)
153
+ log.debug(f"Saved best checkpoint: {filepath}")
153
154
 
154
155
  # Remove worst checkpoint if we've reached save_top_k
155
156
  # NOTE: We add 1 to save_top_k here because we have just saved a new checkpoint
@@ -163,9 +164,9 @@ class BestCheckpoint(Checkpoint):
163
164
  )
164
165
 
165
166
  # Create symlink to best model
166
- _, best_ckpt_path = sorted_ckpts[0]
167
- self._create_symlink(trainer, best_ckpt_path)
168
- log.debug(f"Saved best checkpoint: {filepath}")
167
+ if sorted_ckpts:
168
+ _, best_ckpt_path = sorted_ckpts[0]
169
+ self._create_symlink(trainer, best_ckpt_path)
169
170
 
170
171
  # Barrier to ensure all processes have saved the checkpoint before continuing
171
172
  trainer.strategy.barrier()
{nshtrainer-0.11.4.dist-info → nshtrainer-0.11.5.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.11.4
3
+ Version: 0.11.5
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.11.4.dist-info → nshtrainer-0.11.5.dist-info}/RECORD RENAMED
@@ -11,7 +11,7 @@ nshtrainer/callbacks/_throughput_monitor_callback.py,sha256=aJo_11rc4lo0IYOd-kHm
11
11
  nshtrainer/callbacks/actsave.py,sha256=qbnaKts4_dvjPeAaPtv7Ds12_vEWzaHUfg_--49NB9I,4041
12
12
  nshtrainer/callbacks/base.py,sha256=UnlYZAqSb8UwBJR-N5-XunxFx2yZjZ4lyGqUfhbCRlI,3555
13
13
  nshtrainer/callbacks/checkpoint/__init__.py,sha256=zrEVCGFikfkt0iOMceOFzXsZG2-6QrqY79RKBCS7bu4,738
14
- nshtrainer/callbacks/checkpoint/best_checkpoint.py,sha256=SfM0VGCSDxYutJ21tY8m283eElN8jyKh2hhj6b7-1-s,6121
14
+ nshtrainer/callbacks/checkpoint/best_checkpoint.py,sha256=w99O5GWRcV89XBe4j__v2TvNEHys0x_r3tSTr-6Lhec,6154
15
15
  nshtrainer/callbacks/checkpoint/latest_epoch_checkpoint.py,sha256=NES-acaslPBiZQIMAdk_YwtnBrkm_y_BJQ8Ian0UKP0,4294
16
16
  nshtrainer/callbacks/checkpoint/model_checkpoint.py,sha256=mLFMbNzeMiBer3BCb7o3ucswKpOCQlYyN3wdB92N-LY,6884
17
17
  nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py,sha256=s8tOHrnb_uVqLVeV2K38ZszXrXPTEGdDVfXuXgo_KDQ,3277
@@ -82,6 +82,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
82
82
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
83
83
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
84
84
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
85
- nshtrainer-0.11.4.dist-info/METADATA,sha256=6s-tc2dutPpIQaxv7j8cdZWiLU3l0Y_aS5f-k5JvAfM,860
86
- nshtrainer-0.11.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
87
- nshtrainer-0.11.4.dist-info/RECORD,,
85
+ nshtrainer-0.11.5.dist-info/METADATA,sha256=KHgvYOhQXbc37awWeLbpbdVQbSEU4J7KoC7Lr5286KE,860
86
+ nshtrainer-0.11.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
87
+ nshtrainer-0.11.5.dist-info/RECORD,,