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")