lab07

2023-06-30 17:30:45 +02:00 · 2023-06-30 17:30:45 +02:00 · 7029181457
parent 03a8b1321e
commit 7029181457
1 changed files with 1 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -93,6 +93,7 @@ def my_main(batch_size, learning_rate, epochs):

    model_scripted = torch.jit.script(model) # Export to TorchScript
    model_scripted.save('model_scripted.pt') # Save
+    exint.add_artifact('model_scripted.pt')


 exint.run()