pipeline {
   agent any
   //Definijuemy parametry, ktore bedzie mozna podac podczas wywolywania zadania
   parameters {
     string (
         name: 'epochs_count',
         defaultValue: '10',
         description: 'How many epochs do you want to train?',       
        )
   }
   stages {   
      stage('Run Script') {
            steps {
               script {
                sh 'docker build . -t jhenyk/ium:latest --build-arg epochs=$params.epochs_count'
                sh 'docker run jhenyk/ium:latest'
               }
            }
        
    }
      stage('Archive Output') {
         steps {
            archiveArtifacts 'cifar_net.pth'
         }
      }
   }
}