40 lines
1.1 KiB
Plaintext
40 lines
1.1 KiB
Plaintext
pipeline {
|
|
agent any
|
|
environment {
|
|
dockerImage = ''
|
|
}
|
|
parameters {
|
|
string(name: 'EPOCHS', defaultValue: '100', description: 'Number of epochs')
|
|
string(name: 'LR', defaultValue: '0.01', description: 'Learning rate')
|
|
}
|
|
stages {
|
|
stage('Git Checkout') {
|
|
steps {
|
|
git branch: 'main', url: 'https://git.wmi.amu.edu.pl/s487194/ium_487194'
|
|
}
|
|
}
|
|
stage('Pull Docker image') {
|
|
steps {
|
|
script {
|
|
dockerImage = docker.image('s487194/ium:lab5')
|
|
}
|
|
}
|
|
}
|
|
stage('Train model') {
|
|
steps {
|
|
script {
|
|
dockerImage.inside('-u root') {
|
|
sh "python train.py -epochs ${params.EPOCHS} -lr ${params.LR}"
|
|
}
|
|
}
|
|
}
|
|
}
|
|
stage('Archive model') {
|
|
steps {
|
|
archiveArtifacts artifacts: 'classificationn_model.pt', onlyIfSuccessful: true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|