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)