nshtrainer 1.0.0b53__py3-none-any.whl → 1.0.0b54__py3-none-any.whl

nshtrainer/trainer/trainer.py

@@ -457,7 +457,21 @@ class Trainer(LightningTrainer):
     ):
         filepath = Path(filepath)
 
-        super().save_checkpoint(filepath, weights_only, storage_options)
+        if self.model is None:
+            raise AttributeError(
+                "Saving a checkpoint is only possible if a model is attached to the Trainer. Did you call"
+                " `Trainer.save_checkpoint()` before calling `Trainer.{fit,validate,test,predict}`?"
+            )
+        with self.profiler.profile("save_checkpoint"):  # type: ignore
+            checkpoint = self._checkpoint_connector.dump_checkpoint(weights_only)
+            # Update the checkpoint for the trainer hyperparameters
+            checkpoint[self.CHECKPOINT_HYPER_PARAMS_KEY] = self.hparams.model_dump(
+                mode="json"
+            )
+            self.strategy.save_checkpoint(
+                checkpoint, filepath, storage_options=storage_options
+            )
+            self.strategy.barrier("Trainer.save_checkpoint")
 
         # Save the checkpoint metadata
         metadata_path = None

{nshtrainer-1.0.0b53.dist-info → nshtrainer-1.0.0b54.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: nshtrainer
-Version: 1.0.0b53
+Version: 1.0.0b54
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshtrainer-1.0.0b53.dist-info → nshtrainer-1.0.0b54.dist-info}/RECORD

@@ -142,7 +142,7 @@ nshtrainer/trainer/plugin/layer_sync.py,sha256=h-ydZwXepnsw5-paLgiDatqPyQ_8C0QEv
 nshtrainer/trainer/plugin/precision.py,sha256=I0QsB1bVxmsFmBOkgrAfGONsuYae_lD9Bz0PfJEQvH4,5598
 nshtrainer/trainer/signal_connector.py,sha256=GhfGcSzfaTNhnj2QFkBDq5aT7FqbLMA7eC8SYQs8_8w,10828
 nshtrainer/trainer/strategy.py,sha256=VPTn5z3zvXTydY8IJchjhjcOfpvtoejnvUkq5E4WTus,1368
-nshtrainer/trainer/trainer.py,sha256=QEK-0bcw1y5Cconi99PYFXr0MElUGgGYMZ_SlcJUQ1k,20364
+nshtrainer/trainer/trainer.py,sha256=Lo3vUo3ooTAjaX2fUYPFSMv5FP7sWfVov0QbA-T5hZ8,21113
 nshtrainer/util/_environment_info.py,sha256=MT8mBe6ZolRfKiwU-les1P-lPNPqXpHQcfADrh_A3uY,24629
 nshtrainer/util/bf16.py,sha256=9QhHZCkYSfYpIcxwAMoXyuh2yTSHBzT-EdLQB297jEs,762
 nshtrainer/util/config/__init__.py,sha256=Z39JJufSb61Lhn2GfVcv3eFW_eorOrN9-9llDWlnZZM,272
@@ -154,6 +154,6 @@ nshtrainer/util/seed.py,sha256=diMV8iwBKN7Xxt5pELmui-gyqyT80_CZzomrWhNss0k,316
 nshtrainer/util/slurm.py,sha256=HflkP5iI_r4UHMyPjw9R4dD5AHsJUpcfJw5PLvGYBRM,1603
 nshtrainer/util/typed.py,sha256=Xt5fUU6zwLKSTLUdenovnKK0N8qUq89Kddz2_XeykVQ,164
 nshtrainer/util/typing_utils.py,sha256=MjY-CUX9R5Tzat-BlFnQjwl1PQ_W2yZQoXhkYHlJ_VA,442
-nshtrainer-1.0.0b53.dist-info/METADATA,sha256=xzPw_zTWp8gLpzKhvcAHTv8KbZCvxwMU_MIzMiq8j78,988
-nshtrainer-1.0.0b53.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-nshtrainer-1.0.0b53.dist-info/RECORD,,
+nshtrainer-1.0.0b54.dist-info/METADATA,sha256=oSfrN2tgKgkZJwGbZVNbLULQcVMxh_wb02u7Hrujfn4,988
+nshtrainer-1.0.0b54.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+nshtrainer-1.0.0b54.dist-info/RECORD,,