Update 'train.py'

This commit is contained in:
Patryk Gałka 2023-05-11 21:52:50 +02:00
parent 11f7119d86
commit 360a3901d3

View File

@ -37,7 +37,7 @@ def my_main(EPOCHS, _run):
norm_games_model.fit(games_all, games_predict, epochs=EPOCHS) norm_games_model.fit(games_all, games_predict, epochs=EPOCHS)
norm_games_model.save('test') norm_games_model.save('test')
print(f'done:') _run.add_artifact('test')
exint.run() exint.run()
exint.add_artifact("test") exint.add_artifact("test")