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