From c9c5a261d7f29f6554bc98ef9530db6b15a592c7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= <patgal2@st.amu.edu.pl>
Date: Thu, 11 May 2023 21:55:25 +0200
Subject: [PATCH] Update 'train.py'

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

diff --git a/train.py b/train.py
index c7b5926..3a1dabf 100644
--- a/train.py
+++ b/train.py
@@ -40,5 +40,5 @@ def my_main(EPOCHS, _run):
     _run.add_artifact('test')
 
 exint.run()
-exint.add_artifact("test/")
+exint.add_artifact("test/saved_model.pb")
 exint.add_source_file('./ium_z434686/train.py')