Update 'train.py'

2023-05-11 21:57:16 +02:00 · 2023-05-11 21:57:16 +02:00 · a505cc9813
commit a505cc9813
parent c9c5a261d7
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -40,5 +40,5 @@ def my_main(EPOCHS, _run):
    _run.add_artifact('test')

 exint.run()
-exint.add_artifact("test/saved_model.pb")
+exint.add_artifact(filename="test")
 exint.add_source_file('./ium_z434686/train.py')