nshtrainer 0.29.0__py3-none-any.whl → 0.29.1__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/_base.py +3 -3
  2. {nshtrainer-0.29.0.dist-info → nshtrainer-0.29.1.dist-info}/METADATA +1 -1
  3. {nshtrainer-0.29.0.dist-info → nshtrainer-0.29.1.dist-info}/RECORD +4 -4
  4. {nshtrainer-0.29.0.dist-info → nshtrainer-0.29.1.dist-info}/WHEEL +0 -0
nshtrainer/callbacks/checkpoint/_base.py CHANGED
@@ -155,15 +155,15 @@ class CheckpointBase(Checkpoint, ABC, Generic[TConfig]):
155
155
  trainer.save_checkpoint(filepath, self.config.save_weights_only)
156
156
 
157
157
  if trainer.is_global_zero:
158
+ # Remove old checkpoints
159
+ self.remove_old_checkpoints(trainer)
160
+
158
161
  # Create the latest symlink
159
162
  if (symlink_filename := self.symlink_path()) is not None:
160
163
  symlink_path = self.dirpath / symlink_filename
161
164
  _link_checkpoint(filepath, symlink_path, metadata=True)
162
165
  log.debug(f"Created latest symlink: {symlink_path}")
163
166
 
164
- # Remove old checkpoints
165
- self.remove_old_checkpoints(trainer)
166
-
167
167
  # Barrier to ensure all processes have saved the checkpoint,
168
168
  # deleted the old checkpoints, and created the symlink before continuing
169
169
  trainer.strategy.barrier()
{nshtrainer-0.29.0.dist-info → nshtrainer-0.29.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.29.0
3
+ Version: 0.29.1
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.29.0.dist-info → nshtrainer-0.29.1.dist-info}/RECORD RENAMED
@@ -10,7 +10,7 @@ nshtrainer/callbacks/_throughput_monitor_callback.py,sha256=aJo_11rc4lo0IYOd-kHm
10
10
  nshtrainer/callbacks/actsave.py,sha256=qbnaKts4_dvjPeAaPtv7Ds12_vEWzaHUfg_--49NB9I,4041
11
11
  nshtrainer/callbacks/base.py,sha256=NpjeKmonJ1Kaz5_39XSn3LlDwvbGjk6WV8BpHSNCvI4,3508
12
12
  nshtrainer/callbacks/checkpoint/__init__.py,sha256=g-3zIthupERKqWZQw-A_busQPaPRkto6iHBV-M7nK1Y,527
13
- nshtrainer/callbacks/checkpoint/_base.py,sha256=MzMF7JtvR3A_7DAM2r4NGQSBDisA7krv6WlVk5rKABQ,6157
13
+ nshtrainer/callbacks/checkpoint/_base.py,sha256=vvlwuD-20NozYVIolGGShmUdkkNYeuwN6xCoFnK4GiU,6157
14
14
  nshtrainer/callbacks/checkpoint/best_checkpoint.py,sha256=8BHgLAd3Tuzf5sup0guEAKF1jJiAwYsjdKBFYZw98ac,2171
15
15
  nshtrainer/callbacks/checkpoint/last_checkpoint.py,sha256=CWWv0cSwQ1VAX26N7hAyMxbNCk26Keh39oQguBEK5To,1102
16
16
  nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py,sha256=ctT88EGT22_t_6tr5r7Sfo43cuve6XeroBnBYRMPOus,3372
@@ -87,6 +87,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
87
87
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
88
88
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
89
89
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
90
- nshtrainer-0.29.0.dist-info/METADATA,sha256=EP3cdORGt4w_H0pX-whQJ5ULsO5HQXo3VlHp5bkfqfk,916
91
- nshtrainer-0.29.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
92
- nshtrainer-0.29.0.dist-info/RECORD,,
90
+ nshtrainer-0.29.1.dist-info/METADATA,sha256=Qck1QY1pNnjQH9zLMyAMKVVvYMovEeIyP5zV7VlZios,916
91
+ nshtrainer-0.29.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
92
+ nshtrainer-0.29.1.dist-info/RECORD,,