ium_434766/pytorch-training-eval/Jenkinsfile-train

pipeline {
   agent {
      dockerfile true
   }
   stages {
      stage('checkout') {
            steps {
                git 'https://git.wmi.amu.edu.pl/s434766/ium_434766.git'
                copyArtifacts fingerprintArtifacts: true, projectName: 's434766-create-dataset'
            }
        }
      stage('Docker'){
         steps{
            sh 'python3 "./stroke-pytorch.py" ${BATCH_SIZE} ${EPOCHS} > pred.txt'
         }
      }
      stage('archiveArtifacts') {
         steps {
            archiveArtifacts 'pred.txt'
            archiveArtifacts 'stroke.pkl'
         }
                     post {
                        success {
                           emailext body: 'Training of stroke predictions is finished',
                           subject: 's434766 training finished',
                           to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms'
                }
            }
      }
   }
}