s444018/ium_444018
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

pass epoch to my_main
Some checks failed
s444018-training/pipeline/head There was a failure building this commit
Details
s444018-evaluation/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Szymon Parafiński 2022-05-12 00:07:35 +02:00
parent 356f193c68
commit 3daecf4f3e
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Jenkinsfile_train
View File

@ -20,7 +20,7 @@ pipeline {
} }
stage('Train model with sacred') { stage('Train model with sacred') {
steps { 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 archiveArtifacts artifacts: 'model.pkl, s444018_sacred_FileObserver/**/*.*, result.csv', followSymlinks: false
} }
} }

1
biblioteka_DL/dllib.py
View File

@ -168,6 +168,7 @@ def my_main(epochs, _run):
_run.log_scalar("RMSE", rmse) _run.log_scalar("RMSE", rmse)
_run.log_scalar("MSE", mse) _run.log_scalar("MSE", mse)
_run.info['epochs'] = epochs
# ex.run() # ex.run()
ex.add_artifact("model.pkl") ex.add_artifact("model.pkl")