diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index 5d95677..bc2ea6a 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -3,7 +3,7 @@ pipeline {
         dockerfile  true
     }
     parameters {
-        epochs(
+        string(
             defaultValue: '200',
             description: 'number of epochs',
             name: 'EPOCH'