From 702918145796f6fde286326a5116c39bfdd180ac Mon Sep 17 00:00:00 2001
From: Natalia Szymczyk <natalia.szymczyk@allegro.pl>
Date: Fri, 30 Jun 2023 17:30:45 +0200
Subject: [PATCH] lab07

---
 train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train.py b/train.py
index 4a5f6a8..d827367 100644
--- 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()