mlflow fix

This commit is contained in:
Witold Woch 2023-05-14 17:58:27 +02:00
parent 224fb57583
commit 6aa9dfba3d

View File

@ -13,7 +13,7 @@ pipeline {
stage('Docker and Training') {
steps {
script {
def dockerImage = docker.image('s487194/ium:lab5_7')
def dockerImage = docker.image('s487194/ium:1')
dockerImage.inside {
sh "python train.py --epochs ${params.EPOCHS} --lr ${params.LR}"
archiveArtifacts artifacts: 'classificationn_model.pt', onlyIfSuccessful: true