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