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
|
|
|
}
|
|
|
|
}
|