From aff6e0827b3c9dee952718c80b1de5501875b177 Mon Sep 17 00:00:00 2001
From: Kornelia Girejko <korgir@st.amu.edu.pl>
Date: Fri, 6 May 2022 19:21:59 +0200
Subject: [PATCH] Zaktualizuj 'biblioteka_DL'

---
 biblioteka_DL | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/biblioteka_DL b/biblioteka_DL
index d3cdfc2..1397718 100644
--- a/biblioteka_DL
+++ b/biblioteka_DL
@@ -73,4 +73,6 @@ with torch.no_grad():
     acc = y_predicted_cls.eq(y_testing).sum()/float(y_testing.shape[0])
     print(f'{acc:.4f}')
     result = open("output",'w+')
-    result.write(f'{y_predicted}')
\ No newline at end of file
+    result.write(f'{y_predicted}')
+
+torch.save(model, "model.pkl")
\ No newline at end of file