46 lines
1.0 KiB
Groovy
46 lines
1.0 KiB
Groovy
pipeline {
|
|
agent { dockerfile true }
|
|
|
|
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: 's434695-create-dataset', selector: buildParameter('WHICH_BUILD'))
|
|
}
|
|
}
|
|
stage('train')
|
|
{
|
|
steps
|
|
{
|
|
sh 'python3 train.py ${BATCH_SIZE} ${EPOCHS}'
|
|
}
|
|
}
|
|
|
|
|
|
stage('Archive artifacts') {
|
|
steps{
|
|
archiveArtifacts 'vgsales_model.h5'
|
|
}
|
|
}
|
|
}
|
|
}
|