diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index d08e88a..b699ab8 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:41')
+    def testImage = docker.image('s487197/ium:42')
 	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 2c74042..1242062 100644
--- a/ium_sacred.py
+++ b/ium_sacred.py
@@ -75,7 +75,12 @@ def my_main(epochs, lr, validation_split, _run):
         validation_split=validation_split)
     hist = pd.DataFrame(history.history)
     hist['epoch'] = history.epoch
-
+    print(hist)
+    for his in hist:
+        _run.log_scalar('training.loss', his['loss'])
+        _run.log_scalar('accuracy', his['accuracy'])
+    ex.add_artifact('baltimore_model')
+"""
     baltimore_data_test =pd.read_csv('baltimore_test.csv')
     baltimore_data_test.columns = train.columns
     baltimore_data_test, x_test, y_test = get_x_y(baltimore_data_test)
@@ -101,8 +106,9 @@ def my_main(epochs, lr, validation_split, _run):
     _run.log_scalar('accuracy', data['accuracy'])
     _run.log_scalar('rmse', data['rmse'])
     _run.log_scalar('accuracy', data['accuracy'])
+    """
+
 
-    ex.add_artifact('baltimore_model')
 
 ex.run()