From d692b377ecede7266b9d26dca34244c61cced1b6 Mon Sep 17 00:00:00 2001
From: NM512 <morihira3513@gmailcom>
Date: Fri, 5 May 2023 01:32:08 +0900
Subject: [PATCH] memory saving at evaluation

---
 dreamer.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/dreamer.py b/dreamer.py
index ce6152c..e4abb6f 100644
--- a/dreamer.py
+++ b/dreamer.py
@@ -256,12 +256,12 @@ class ProcessEpisodeWrap:
             # use dataset_size as log step for a condition of envs > 1
             log_step = total * config.action_repeat
         elif mode == "eval":
-            # start saving episodes for evaluation
+            # keep only last item for saving memory
+            while len(cache) > 1:
+                # FIFO
+                cache.popitem()
+            # start counting scores for evaluation
             if cls.last_step_at_eval != logger.step:
-                # keep only last item
-                while len(cache) > 1:
-                    # FIFO
-                    cache.popitem()
                 cls.eval_scores = []
                 cls.eval_lengths = []
                 cls.eval_done = False