From 8d5658d672bb66e3bee7df792b14a1829c52bf41 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= <sebwal1@st.amu.edu.pl>
Date: Fri, 6 May 2022 19:23:57 +0200
Subject: [PATCH] add torch.save model

---
 ml_pytorch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ml_pytorch.py b/ml_pytorch.py
index 9d59a7b..826b006 100644
--- a/ml_pytorch.py
+++ b/ml_pytorch.py
@@ -156,4 +156,4 @@ with open("result.txt", "w+") as file:
         input_, target = val_ds[i]
         file.write(str(predict_single(input_, target, model)))
 
-
+torch.save(model, "Model_xPosition.pkl")