ium_434760/Jenkinsfile-train

46 lines
1.2 KiB
Plaintext
Raw Normal View History

2021-04-28 20:21:08 +02:00
pipeline {
agent {
2021-05-24 01:00:39 +02:00
docker { image 'adnovac/ium_s434760:3.1' }
2021-04-28 20:21:08 +02:00
}
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 {
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-05-24 00:52:24 +02:00
archiveArtifacts 'fifa_overall/**/*'
2021-04-28 21:38:41 +02:00
}
2021-04-28 20:21:08 +02:00
}
}
}