From f7ed22c82e2939625538cca0aa631bc398c40c2a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= Date: Thu, 11 May 2023 22:19:07 +0200 Subject: [PATCH] Update 'train.py' --- train.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index 838afed..82af62d 100644 --- a/train.py +++ b/train.py @@ -39,5 +39,7 @@ def my_main(EPOCHS, _run): norm_games_model.save('test') exint.run() -exint.add_artifact(filename="test/saved_model.pb") +exint.add_artifact(filename="test/*") +exint.add_artifact(filename="metrics.csv") +exint.add_artifact(filename="metrics.png") exint.add_source_file('./ium_z434686/train.py')