training inside docker

2021-05-07 18:55:51 +02:00 · 2021-05-07 18:55:51 +02:00 · 367ec7d511
commit 367ec7d511
parent a53d4773fc
1 changed files with 15 additions and 10 deletions
--- a/25
+++ b/25
@ -1,9 +1,6 @@
 pipeline {
-    agent {
-        docker {
-            image 'patlaz/ium:1.0'
-        }
-    }	
+    agent any
+
 	parameters {
 		buildSelector(
 					defaultSelector: lastSuccessful(),
@ -27,14 +24,22 @@ pipeline {
                copyArtifacts fingerprintArtifacts: true, projectName: 's434742-create-dataset', selector: buildParameter('BUILD_SELECTOR')
            }
        }
-		stage('training') {
+
+        stage('docker-training') {
 			steps {
-                sh 'chmod +x avocado-training.py'
-				sh 'python3 avocado-training.py ${EPOCHS} ${BATCH_SIZE}'
-                
-			
+                script {
+                    def img = docker.build('patlaz/ium:1.0')
+                    img.inside {
+                        sh 'chmod +x avocado-training.py'
+                        sh 'echo ${EPOCHS} ${BATCH_SIZE}'
+                        sh 'python3 avocado-training.py ${EPOCHS} ${BATCH_SIZE}'
+                 
+                    }
+                }
 			}
 		}
+
+
 		stage('archiveArtifacts') {
 			steps{
 				archiveArtifacts 'avocado-model.h5'