From 8f958c000e2e9a3a8847b93f9e89ecea0922f377 Mon Sep 17 00:00:00 2001 From: Alagris Date: Mon, 26 Apr 2021 08:25:18 +0200 Subject: [PATCH] archvie results --- train-model.Jenkinsfile | 2 +- train_model.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/train-model.Jenkinsfile b/train-model.Jenkinsfile index e36ed8b..cdbb94e 100644 --- a/train-model.Jenkinsfile +++ b/train-model.Jenkinsfile @@ -13,7 +13,7 @@ pipeline { post { success { - archiveArtifacts 'cnn.pth' + archiveArtifacts 'cnn.pth,results.txt' } } } diff --git a/train_model.py b/train_model.py index 8f6bf60..9aee84b 100644 --- a/train_model.py +++ b/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) - print("Average estimation error " + str(diff.item() / i)) + 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)