ium_434760/Jenkinsfile-train
2021-04-28 22:46:19 +02:00

53 lines
1.5 KiB
Plaintext

pipeline {
agent {
docker { image 'adnovac/ium_s434760:latest' }
}
parameters{
buildSelector(
defaultSelector: lastSuccessful(),
description: 'Which build to use for copying artifacts',
name: 'WHICH_BUILD'
)
string(
defaultValue: '16',
description: 'batch size',
name: 'BATCH_SIZE'
)
string(
defaultValue: '15',
description: 'epochs',
name: 'EPOCHS'
)
}
stages {
stage('copy artifacts')
{
steps
{
copyArtifacts(fingerprintArtifacts: true, projectName: 's434760-create-dataset', selector: buildParameter('WHICH_BUILD'))
}
}
stage('train')
{
steps
{
catchError {
sh 'pip3 install -r requirements.txt'
sh 'python3.8 train.py ${BATCH_SIZE} ${EPOCHS}'
}
}
}
stage('archive artifacts') {
steps {
archiveArtifacts 'model.h5'
}
}
stage('send email') {
steps {
emailext body: currentBuild.result,
subject: 's434760 - train',
to: 'annnow19@st.amu.edu.pl'
}
}
}
}