From ce684a4468fe8f3e20ad92a5067f6c375f7c72a4 Mon Sep 17 00:00:00 2001
From: Wojciech Lidwin <59863785+Halal37@users.noreply.github.com>
Date: Fri, 12 May 2023 04:28:14 +0200
Subject: [PATCH] Fix

---
 Jenkinsfile_train | 2 +-
 ium_sacred.py     | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index b8baddf..e47705d 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -25,7 +25,7 @@ node {
      checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's487197', url: 'https://git.wmi.amu.edu.pl/s487197/ium_487197']]])
  }
  stage('Dockerfile'){
-    def testImage = docker.image('s487197/ium:46')
+    def testImage = docker.image('s487197/ium:47')
 	testImage.inside{ 
          copyArtifacts filter: 'baltimore_train.csv', projectName: 's487197-create-dataset'
 		 sh "python3 ium_sacred.py -epochs $EPOCHS -lr $LR -validation_split $VALIDATION_SPLIT"
diff --git a/ium_sacred.py b/ium_sacred.py
index ff18e32..b8e5e2f 100644
--- a/ium_sacred.py
+++ b/ium_sacred.py
@@ -79,6 +79,7 @@ def my_main(epochs, lr, validation_split, _run):
     for his in hist.iterrows():
         _run.log_scalar('training.loss', his[1]['loss'])
         _run.log_scalar('accuracy', his[1]['accuracy'])
+    model.save('baltimore_model')
     ex.add_artifact('baltimore_model')
 
 """
@@ -109,7 +110,5 @@ def my_main(epochs, lr, validation_split, _run):
     _run.log_scalar('accuracy', data['accuracy'])
     """
 
-
-
-ex.run()
+ex.run_commandline()