clear eval episodes for saving memory

2023-03-20 20:55:06 +09:00 · 2023-03-20 20:55:06 +09:00 · 5ad0f6e9ca
commit 5ad0f6e9ca
parent 44f8edc115
1 changed files with 1 additions and 0 deletions
--- 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