fix: ignore prototype_win_ratios by loading with strict=False

2022-02-02 21:52:01 +01:00
parent 197b728c63
commit 15e7232747
2 changed files with 6 additions and 4 deletions
--- a/examples/glvq_iris.py
+++ b/examples/glvq_iris.py
@@ -59,5 +59,7 @@ if __name__ == "__main__":

    # Load saved model
    new_model = pt.models.GLVQ.load_from_checkpoint(
-        checkpoint_path="./glvq_iris.ckpt")
+        checkpoint_path="./glvq_iris.ckpt",
+        strict=False,
+    )
    print(new_model)