mlflow fix
This commit is contained in:
parent
224fb57583
commit
6aa9dfba3d
@ -13,7 +13,7 @@ pipeline {
|
|||||||
stage('Docker and Training') {
|
stage('Docker and Training') {
|
||||||
steps {
|
steps {
|
||||||
script {
|
script {
|
||||||
def dockerImage = docker.image('s487194/ium:lab5_7')
|
def dockerImage = docker.image('s487194/ium:1')
|
||||||
dockerImage.inside {
|
dockerImage.inside {
|
||||||
sh "python train.py --epochs ${params.EPOCHS} --lr ${params.LR}"
|
sh "python train.py --epochs ${params.EPOCHS} --lr ${params.LR}"
|
||||||
archiveArtifacts artifacts: 'classificationn_model.pt', onlyIfSuccessful: true
|
archiveArtifacts artifacts: 'classificationn_model.pt', onlyIfSuccessful: true
|
||||||
|
Loading…
Reference in New Issue
Block a user