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