diff --git a/Jenkinsfile-training b/Jenkinsfile-training
index 08c8357..72b62ff 100644
--- a/Jenkinsfile-training
+++ b/Jenkinsfile-training
@@ -1,6 +1,14 @@
 pipeline{
     agent any
 
+    parameters {
+        string(
+            name: 'EPOCHS',
+            description: 'Number of epochs',
+            defaultValue: '10'
+        )
+    }
+
     stages{
         stage('Copy dataset'){
             steps{
@@ -22,12 +30,4 @@ pipeline{
             }
         }
     }
-
-    parameters{
-        string{
-            name: 'EPOCHS',
-            description: 'Number of epochs',
-            defaultValue: '10',
-        }
-    }
 }
\ No newline at end of file