From ceb1643cc0672febd0b1e28e503b21c49d60b9a6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20=C5=81=C4=85czkowski?= <broketmpl@gmail.com>
Date: Fri, 26 Apr 2024 09:41:35 +0200
Subject: [PATCH] IUM_07 - fix models/Jenkinsfile

---
 models/Jenkinsfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/models/Jenkinsfile b/models/Jenkinsfile
index 0f78674..37bd456 100644
--- a/models/Jenkinsfile
+++ b/models/Jenkinsfile
@@ -52,6 +52,8 @@ pipeline {
                     sh "chmod +x ./create_model.py"
                     sh "python3 ./create_model.py with 'num_epochs=${params.epochs}' 'learning_rate=${params.learning_rate}' 'weight_decay=${params.weight_decay}'"
 
+                    archiveArtifacts artifacts: "models/model.pth", onlyIfSuccessful: true
+
                     def experiment_id = readFile('experiment_id.txt').trim()
                     archiveArtifacts artifacts: "my_runs/${experiment_id}/*", onlyIfSuccessful: true