Zaktualizuj 'ml_pytroch_sacred.py'

2022-05-09 11:27:48 +02:00 · 2022-05-09 11:27:48 +02:00 · 94f58c9fb9
commit 94f58c9fb9
parent 06f03475f0
1 changed files with 2 additions and 1 deletions
--- a/ml_pytroch_sacred.py
+++ b/ml_pytroch_sacred.py
@ -127,4 +127,5 @@ def my_main(epochs):
            input_, target = val_ds[i]
            file.write(str(predict_single(input_, target, model)))
    
-    torch.save(model, "Model_xPosition.pkl")
+    torch.save(model, "Model_xPosition.pkl")
+    ex.add_artifact("Model_xPosition.pkl")