From 94f58c9fb9707bf35574eb6478ddb71bd03958ff Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Mon, 9 May 2022 11:27:48 +0200 Subject: [PATCH] Zaktualizuj 'ml_pytroch_sacred.py' --- ml_pytroch_sacred.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ml_pytroch_sacred.py b/ml_pytroch_sacred.py index 415d37a..ffbc818 100644 --- 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") \ No newline at end of file + torch.save(model, "Model_xPosition.pkl") + ex.add_artifact("Model_xPosition.pkl") \ No newline at end of file