diff --git a/evaluation.Jenkinsfile b/evaluation.Jenkinsfile index b470c9f..48d4d00 100644 --- a/evaluation.Jenkinsfile +++ b/evaluation.Jenkinsfile @@ -2,13 +2,17 @@ pipeline { agent { dockerfile true } - + parameters { + string(name: 'batch_size', defaultValue: '512', description: 'size of batch') + string(name: 'learning_rate', defaultValue: '0.01', description: 'Learning rate') + string(name: 'epochs', defaultValue: '14', description: 'epochs') + } stages { stage('Build') { steps { git 'https://git.wmi.amu.edu.pl/s434749/ium_434749.git' copyArtifacts fingerprintArtifacts: true, projectName: 's434749-training', selector: lastSuccessful() - sh 'python3 train_model.py with "mode=eval"' + sh "python3 train_model.py with 'mode=eval' 'batch_size=${params.batch_size}' 'learning_rate=${params.learning_rate}' 'epochs=${params.epochs}'" script{ def results = readFile "${env.WORKSPACE}/results.txt" } diff --git a/train-model.Jenkinsfile b/train-model.Jenkinsfile index 4f129d0..4507346 100644 --- a/train-model.Jenkinsfile +++ b/train-model.Jenkinsfile @@ -2,13 +2,17 @@ pipeline { agent { dockerfile true } - + parameters { + string(name: 'batch_size', defaultValue: '512', description: 'size of batch') + string(name: 'learning_rate', defaultValue: '0.01', description: 'Learning rate') + string(name: 'epochs', defaultValue: '14', description: 'epochs') + } stages { stage('Build') { steps { git 'https://git.wmi.amu.edu.pl/s434749/ium_434749.git' copyArtifacts fingerprintArtifacts: true, projectName: 's434749-create-dataset', selector: lastSuccessful() - sh 'python3 train_model.py' + sh "python3 train_model.py with 'batch_size=${params.batch_size}' 'learning_rate=${params.learning_rate}' 'epochs=${params.epochs}'" } post {