Ver código fonte

Fixed results writing

Nicholas Schense 6 dias atrás
pai
commit
38117a2590
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      train_model.py

+ 1 - 1
train_model.py

@@ -130,7 +130,7 @@ for run_num in range(config["training"]["ensemble_size"]):
 
     # Save test results by appending to the results file
     test_results_save_path = pl.Path(config["output"]["path"]) / f"results.json"
-    with open(test_results_save_path, "wr+") as f:
+    with open(test_results_save_path, "r+") as f:
         try:
             results = json.load(f)
         except json.JSONDecodeError: