s434749/ium_434749
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix
All checks were successful
s123456-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Alagris 2021-04-26 10:45:58 +02:00
parent 8f958c000e
commit 6853d49f20
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_model.py
View File

@ -165,7 +165,7 @@ def evaluate_monte_carlo(model, repeats):
i += BATCH_SIZE i += BATCH_SIZE
inner_bar.update(2 * BATCH_SIZE) inner_bar.update(2 * BATCH_SIZE)
outer_bar.update(1) 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)) print("Average estimation error " + str(diff.item() / i))
r.write("Average estimation error " + str(diff.item() / i)+"\n") r.write("Average estimation error " + str(diff.item() / i)+"\n")