fix

2021-04-26 10:45:58 +02:00 · 2021-04-26 10:45:58 +02:00 · 6853d49f20
commit 6853d49f20
parent 8f958c000e
1 changed files with 1 additions and 1 deletions
--- 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")