Update 'train.py'

2023-05-11 21:52:50 +02:00 · 2023-05-11 21:52:50 +02:00 · 360a3901d3
commit 360a3901d3
parent 11f7119d86
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -37,7 +37,7 @@ def my_main(EPOCHS, _run):
    norm_games_model.fit(games_all, games_predict, epochs=EPOCHS)

    norm_games_model.save('test')
-    print(f'done:')
+    _run.add_artifact('test')

 exint.run()
 exint.add_artifact("test")