Update 'train.py'

This commit is contained in:
Patryk Gałka 2023-05-11 22:03:52 +02:00
parent a505cc9813
commit 1795e64e25

View File

@ -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')