Idiotproof file operations

2022-05-05 23:22:10 +02:00 · 2022-05-05 23:22:10 +02:00 · 5b2b990ccb
commit 5b2b990ccb
parent 5f4b34648d
1 changed files with 2 additions and 4 deletions
--- a/eval_model.py
+++ b/eval_model.py
@ -31,11 +31,9 @@ def main():
    loss_fn = torch.nn.MSELoss()

    loss = test(dataloader, model, loss_fn)
-    with open('evaluation_results.txt', 'r+') as f:
-        f.read()
+    with open('evaluation_results.txt', 'a+') as f:
        f.write(f'{str(loss)}\n')
-        f.flush()
-        f.seek(0)
+    with open('evaluation_results.txt', 'r') as f:   
        values = [float(line) for line in f.readlines() if line]
        make_plot(values)