From 1795e64e253cb657410a8f315c13ca76f10a36bc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= Date: Thu, 11 May 2023 22:03:52 +0200 Subject: [PATCH] Update 'train.py' --- train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/train.py b/train.py index 2c82a6b..2ce55a7 100644 --- a/train.py +++ b/train.py @@ -37,8 +37,7 @@ def my_main(EPOCHS, _run): norm_games_model.fit(games_all, games_predict, epochs=EPOCHS) norm_games_model.save('test') - _run.add_artifact('test') exint.run() -exint.add_artifact(filename="test") +exint.add_artifact(filename="test/saved_model.pb") exint.add_source_file('./ium_z434686/train.py')