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