ium_434760/Jenkinsfile-train

53 lines
1.5 KiB
Plaintext
Raw Normal View History

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