From 6853d49f20765cc6d775658b47e5df4a1c19604d Mon Sep 17 00:00:00 2001 From: Alagris Date: Mon, 26 Apr 2021 10:45:58 +0200 Subject: [PATCH] fix --- train_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_model.py b/train_model.py index 9aee84b..5b6a642 100644 --- a/train_model.py +++ b/train_model.py @@ -165,7 +165,7 @@ 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: + with open('results.txt', 'w+') as r: print("Average estimation error " + str(diff.item() / i)) r.write("Average estimation error " + str(diff.item() / i)+"\n")