add EPOCHS variable

2022-05-06 20:19:11 +02:00 · 2022-05-06 20:19:11 +02:00 · 2ea7562953
commit 2ea7562953
parent 7dd45b9c75
1 changed files with 9 additions and 1 deletions
--- a/10
+++ b/10
@ -4,11 +4,19 @@ pipeline {
 			 additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=${params.KAGGLE_KEY} --build-arg CUTOFF=${params.CUTOFF} -t sebastianwalesa"
        }
    }
+    parameters {
+        string(
+            defaultValue: '1000',
+            description: 'Number of epochs',
+            name: 'EPOCHS',
+            trim: false
+        )
+    }
    stages {
        stage('Train-stage'){
 		steps {
                copyArtifacts filter: '*', projectName: 's478839-create-dataset'
-			    sh 'python3 ./ml_pytorch.py'
+			    sh 'python3 ./ml_pytorch.py $EPOCHS'
 			    archiveArtifacts artifacts: 'Model_xPosition.pkl', followSymlinks: false
 		}
 	}