pass epoch to my_main

2022-05-12 00:07:35 +02:00 · 2022-05-12 00:07:35 +02:00 · 3daecf4f3e
commit 3daecf4f3e
parent 356f193c68
2 changed files with 2 additions and 1 deletions
--- a/2
+++ b/2
@ -20,7 +20,7 @@ pipeline {
 	   }
 	stage('Train model with sacred') {
 	    steps {
-	        sh 'python3 ./biblioteka_DL/dllib.py $EPOCHS'
+	        sh 'python3 ./biblioteka_DL/dllib.py with "epochs=$EPOCHS"''
            archiveArtifacts artifacts: 'model.pkl, s444018_sacred_FileObserver/**/*.*, result.csv', followSymlinks: false
           }
        }
--- a/biblioteka_DL/dllib.py
+++ b/biblioteka_DL/dllib.py
@ -168,6 +168,7 @@ def my_main(epochs, _run):

    _run.log_scalar("RMSE", rmse)
    _run.log_scalar("MSE", mse)
+    _run.info['epochs'] = epochs

 # ex.run()
 ex.add_artifact("model.pkl")