hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

memory saving at evaluation

Browse Source
This commit is contained in:
NM512 2023-05-05 01:32:08 +09:00
parent e5e8bcb284
commit d692b377ec
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
dreamer.py
View File

@ -256,12 +256,12 @@ class ProcessEpisodeWrap:
# use dataset_size as log step for a condition of envs > 1 # use dataset_size as log step for a condition of envs > 1
log_step = total * config.action_repeat log_step = total * config.action_repeat
elif mode == "eval": 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: if cls.last_step_at_eval != logger.step:
# keep only last item
while len(cache) > 1:
# FIFO
cache.popitem()
cls.eval_scores = [] cls.eval_scores = []
cls.eval_lengths = [] cls.eval_lengths = []
cls.eval_done = False cls.eval_done = False