diff --git a/.gitignore b/.gitignore index 14a44bc..6e381fa 100644 --- a/.gitignore +++ b/.gitignore @@ -156,4 +156,6 @@ Car_Prices_Poland_Kaggle* CarPrices* IUM08/* .DS_store -*.db \ No newline at end of file +*.db +mlruns +my_model \ No newline at end of file diff --git a/Jenkinsfile_training b/Jenkinsfile_training index fff2cd1..fdf7b4a 100644 --- a/Jenkinsfile_training +++ b/Jenkinsfile_training @@ -9,9 +9,14 @@ pipeline { string(name: 'epoch', defaultValue: '100', description: 'Number of epochs to train model.') } stages { - stage('Get arifacts') { + stage('Get artifacts') { steps { copyArtifacts fingerprintArtifacts: true, projectName: 's444507-create-dataset', selector: lastSuccessful() + } + } + stage('Run mlflow script and save artifacts') { + steps { + sh "python3 lab08_deepLearining_mlflow.py $epoch" archiveArtifacts artifacts: 'CarPrices_pytorch_model.pkl' archiveArtifacts artifacts: 'mlruns/**' archiveArtifacts artifacts: 'my_model/**' @@ -19,11 +24,6 @@ pipeline { sh 'rm -r my_model' } } - stage('Train model with sacred') { - steps { - sh "python3 lab08_deepLearining_mlflow.py $epoch" - } - } } post { always {