From 391c7e5d59962429fb4d709ea751b05704d76d62 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Sun, 8 May 2022 19:32:54 +0200 Subject: [PATCH] Zaktualizuj 'ml_pytroch_sacred.py' --- ml_pytroch_sacred.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ml_pytroch_sacred.py b/ml_pytroch_sacred.py index fff8fa0..9e7b7bd 100644 --- a/ml_pytroch_sacred.py +++ b/ml_pytroch_sacred.py @@ -122,4 +122,6 @@ def my_main(epochs): with open("result.txt", "w+") as file: for i in range(0, len(val_ds), 1): input_, target = val_ds[i] - file.write(str(predict_single(input_, target, model))) \ No newline at end of file + file.write(str(predict_single(input_, target, model))) + + torch.save(model, "Model_xPosition.pkl") \ No newline at end of file