Update 'train.py'
This commit is contained in:
parent
a505cc9813
commit
1795e64e25
3
train.py
3
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')
|
||||
|
Loading…
Reference in New Issue
Block a user