nshtrainer 1.0.0b53__tar.gz → 1.0.0b54__tar.gz

{nshtrainer-1.0.0b53 → nshtrainer-1.0.0b54}/PKG-INFO

@@ -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 → nshtrainer-1.0.0b54}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshtrainer"
-version = "1.0.0-beta53"
+version = "1.0.0-beta54"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"

{nshtrainer-1.0.0b53 → nshtrainer-1.0.0b54}/src/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