s487194/ium_487194
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix5

Browse Source
This commit is contained in:
Witold Woch 2023-05-12 20:46:27 +02:00
parent 693e4729d9
commit d0f661d8ef
1 changed files with 4 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
Jenkins_train
View File

@ -1,8 +1,5 @@
pipeline { pipeline {
agent any agent any
environment {
dockerImage = ''
}
parameters { parameters {
string(name: 'EPOCHS', defaultValue: '100', description: 'Number of epochs') string(name: 'EPOCHS', defaultValue: '100', description: 'Number of epochs')
string(name: 'LR', defaultValue: '0.01', description: 'Learning rate') string(name: 'LR', defaultValue: '0.01', description: 'Learning rate')
@ -13,28 +10,16 @@ pipeline {
checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's487194', url: 'https://git.wmi.amu.edu.pl/s487194/ium_487194']]]) checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's487194', url: 'https://git.wmi.amu.edu.pl/s487194/ium_487194']]])
} }
} }
stage('Pull Docker image') { stage('Docker and Training'){
steps { steps {
script { script {
dockerImage = docker.image('s487194/ium:lab5') def dockerImage = docker.image('s487194/ium:lab5')
} dockerImage.inside {
}
}
stage('Train model') {
steps {
script {
dockerImage.inside('-u root') {
sh "python train.py -epochs ${params.EPOCHS} -lr ${params.LR}" sh "python train.py -epochs ${params.EPOCHS} -lr ${params.LR}"
}
}
}
}
stage('Archive model') {
steps {
archiveArtifacts artifacts: 'classificationn_model.pt', onlyIfSuccessful: true archiveArtifacts artifacts: 'classificationn_model.pt', onlyIfSuccessful: true
} }
} }
} }
} }
}
}