Log to artifacts

2022-05-11 20:05:31 +02:00 · 2022-05-11 20:05:31 +02:00 · 1679fac800
commit 1679fac800
parent e93d644059
2 changed files with 4 additions and 1 deletions
--- a/3
+++ b/3
@ -36,6 +36,9 @@ pipeline {
        stage('Train model') {
            steps {
                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') {
--- a/train_model.py
+++ b/train_model.py
@ -14,7 +14,7 @@ default_epochs = 5

 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")