chore: add example for saving/loading models from checkpoints

2022-02-02 19:02:26 +01:00
parent d5855dbe97
commit 98892afee0
1 changed files with 8 additions and 0 deletions
--- a/examples/glvq_iris.py
+++ b/examples/glvq_iris.py
@@ -53,3 +53,11 @@ if __name__ == "__main__":

    # Training loop
    trainer.fit(model, train_loader)
+
+    # Manual save
+    trainer.save_checkpoint("./glvq_iris.ckpt")
+
+    # Load saved model
+    new_model = pt.models.GLVQ.load_from_checkpoint(
+        checkpoint_path="./glvq_iris.ckpt")
+    print(new_model)