archvie results

train-model.Jenkinsfile

@@ -13,7 +13,7 @@ pipeline {
 
             post {
                 success {
-                    archiveArtifacts 'cnn.pth'
+                    archiveArtifacts 'cnn.pth,results.txt'
                 }
             }
         }

train_model.py

@@ -165,7 +165,9 @@ def evaluate_monte_carlo(model, repeats):
                 i += BATCH_SIZE
                 inner_bar.update(2 * BATCH_SIZE)
             outer_bar.update(1)
+        with open('results.txt') as r:
             print("Average estimation error " + str(diff.item() / i))
+            r.write("Average estimation error " + str(diff.item() / i)+"\n")
 
 
 cnn = CNN(kernel_size=3, hidden_layers=14, channels=MAX_LEN, embedding_size=MAX_LEN).to(DEVICE)