From 5ad0f6e9ca0ab483e250e0108a8def9be00849db Mon Sep 17 00:00:00 2001
From: NM512 <morihira3513@gmailcom>
Date: Mon, 20 Mar 2023 20:55:06 +0900
Subject: [PATCH] clear eval episodes for saving memory

---
 dreamer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dreamer.py b/dreamer.py
index eff53ce..79951b7 100644
--- a/dreamer.py
+++ b/dreamer.py
@@ -247,6 +247,7 @@ class ProcessEpisodeWrap:
                 episode_num = len(cls.eval_scores)
                 cls.eval_scores = []
                 cls.eval_lengths = []
+                cache.clear()
 
         if mode == "train" and config.dataset_size:
             total = 0