Merge pull request #49 from JulesBelveze/fix/memory-overflow

[tsa/eval] - fix: ensure matplotlib figures are closed after saving
JulesBelveze · Oct 27, 2024 · 26cffc0 · 26cffc0
2 parents b5bc494 + 4cb8c0c
commit 26cffc0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tsa/eval.py b/tsa/eval.py
@@ -50,6 +50,7 @@ def evaluate(test_iter, criterion, model, config, ts):
         plt.plot(preds, linewidth=.3)
         plt.plot(targets, linewidth=.3)
         plt.savefig("{}/preds.png".format(config.general.output_dir))
+        plt.close()
 
         torch.save(targets, os.path.join(config.general.output_dir, "targets.pt"))
         torch.save(predictions, os.path.join(config.general.output_dir, "predictions.pt"))