47 lines
1.3 KiB
Plaintext
47 lines
1.3 KiB
Plaintext
|
pipeline {
|
||
|
|
agent any
|
||
|
|
parameters {
|
||
|
|
buildSelector(
|
||
|
|
name: 'BUILD_SELECTOR',
|
||
|
|
defaultSelector: lastSuccessful(),
|
||
|
|
description: 'A build to take the artifacts from'
|
||
|
|
)
|
||
|
|
string(
|
||
|
|
name: 'EPOCHS',
|
||
|
|
description: 'Number of epochs',
|
||
|
|
defaultValue: '10'
|
||
|
|
)
|
||
|
|
}
|
||
|
|
stages {
|
||
|
|
stage('Copy artifacts') {
|
||
|
|
steps {
|
||
|
|
script {
|
||
|
|
copyArtifacts(
|
||
|
|
projectName: 'z-s487179-create-dataset',
|
||
|
|
selector: buildParameter('BUILD_SELECTOR'),
|
||
|
|
target: './MLTrain'
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
stage('Run training and save model') {
|
||
|
|
steps {
|
||
|
|
script {
|
||
|
|
sh 'ls -l'
|
||
|
|
docker.image('docker-image').inside {
|
||
|
|
dir('./MLTrain') {
|
||
|
|
sh 'ls -l'
|
||
|
|
sh 'python3 ./model_train.py'
|
||
|
|
archiveArtifacts 'model.pt'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
post {
|
||
|
|
success {
|
||
|
|
build job: 'z-s487179-evaluation.eg/main', propagate: false, wait: false
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|