s444409/ium_444409
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Log to artifacts
All checks were successful
s444409-evaluation/pipeline/head This commit looks good
Details
s444409-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Marcin Kostrzewski 2022-05-11 20:05:31 +02:00
parent e93d644059
commit 1679fac800
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Jenkinsfile-train
View File

@ -36,6 +36,9 @@ pipeline {
stage('Train model') { stage('Train model') {
steps { steps {
sh "python train_model.py -e ${params.EPOCHS} -b ${params.BATCHSIZE}" sh "python train_model.py -e ${params.EPOCHS} -b ${params.BATCHSIZE}"
archiveArtifacts artifacts: 'model_out', onlyIfSuccessful: true
archiveArtifacts artifacts: 'mlruns/**', onlyIfSuccessful: true
sh 'rm -r mlruns'
} }
} }
stage('Evaluate model') { stage('Evaluate model') {

2
train_model.py
View File

@ -14,7 +14,7 @@ default_epochs = 5
device = "cuda" if torch.cuda.is_available() else "cpu" device = "cuda" if torch.cuda.is_available() else "cpu"
mlflow.set_tracking_uri("http://172.17.0.1:5000") # mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_experiment("s444409") mlflow.set_experiment("s444409")