Update 'train.py'

This commit is contained in:
Patryk Gałka 2023-05-11 22:25:16 +02:00
parent b57e04c378
commit 608c4377e5

View File

@ -40,6 +40,6 @@ def my_main(EPOCHS, _run):
exint.run() exint.run()
exint.add_artifact(filename="test/saved_model.pb") exint.add_artifact(filename="test/saved_model.pb")
exint.add_artifact(filename="metrics.csv") exint.add_artifact(filename="./ium_z434686/metrics.csv")
exint.add_artifact(filename="metrics.png") exint.add_artifact(filename="./ium_z434686/metrics.png")
exint.add_source_file('./ium_z434686/train.py') exint.add_source_file('./ium_z434686/train.py')