From 93a44207552482332990b5ea9982ae18f6c5ae3a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20=C5=81=C4=85czkowski?= <broketmpl@gmail.com>
Date: Wed, 24 Apr 2024 22:01:29 +0200
Subject: [PATCH] IUM_07 - fix models/Jenkinsfile, update create_model.py

---
 create_model.py    | 6 ------
 models/Jenkinsfile | 2 +-
 2 files changed, 1 insertion(+), 7 deletions(-)

diff --git a/create_model.py b/create_model.py
index 554f57c..80bdce1 100644
--- a/create_model.py
+++ b/create_model.py
@@ -33,12 +33,6 @@ def config():
     weight_decay = 0.001
     num_epochs = 1000
 
-    # Learning parameters from sys.argv
-    if len(sys.argv) > 1:
-        num_epochs = int(sys.argv[1])
-        learning_rate = float(sys.argv[2])
-        weight_decay = float(sys.argv[3])
-
 @ex.automain
 def experiment(hidden_size, learning_rate, weight_decay, num_epochs, _run):
     # Seed for reproducibility
diff --git a/models/Jenkinsfile b/models/Jenkinsfile
index 952b520..21e4425 100644
--- a/models/Jenkinsfile
+++ b/models/Jenkinsfile
@@ -50,7 +50,7 @@ pipeline {
             steps {
                 script {
                     sh "chmod +x ./create_model.py"
-                    sh "python3 ./create_model.py with 'epochs=${params.epochs}' 'learning_rate=${params.learning_rate}' 'weight_decay=${params.weight_decay}'"
+                    sh "python3 ./create_model.py with 'num_epochs=${params.epochs}' 'learning_rate=${params.learning_rate}' 'weight_decay=${params.weight_decay}'"
 
                     def experiment_id = readFile('experiment_id.txt').trim()
                     archiveArtifacts artifacts: "my_runs/${experiment_id}/*", onlyIfSuccessful: true