s478839/ium_478839
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add EPOCHS variable

Browse Source
This commit is contained in:
Sebastian Wałęsa 2022-05-06 20:19:11 +02:00
parent 7dd45b9c75
commit 2ea7562953
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Jenkinsfile-train
View File

@ -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" 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 { stages {
stage('Train-stage'){ stage('Train-stage'){
steps { steps {
copyArtifacts filter: '*', projectName: 's478839-create-dataset' copyArtifacts filter: '*', projectName: 's478839-create-dataset'
sh 'python3 ./ml_pytorch.py' sh 'python3 ./ml_pytorch.py $EPOCHS'
archiveArtifacts artifacts: 'Model_xPosition.pkl', followSymlinks: false archiveArtifacts artifacts: 'Model_xPosition.pkl', followSymlinks: false
} }
} }