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