hongshaorou/tdmpc2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cache buffer values in offline training

Browse Source
This commit is contained in:
Nicklas Hansen
2024-12-19 09:40:04 -08:00
parent 9cac7c5775
commit 66f8c21f58
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tdmpc2/common/buffer.py
View File

@@ -20,6 +20,7 @@ class Buffer():
traj_key='episode', traj_key='episode',
truncated_key=None, truncated_key=None,
strict_length=True, strict_length=True,
cache_values=cfg.multitask,
) )
self._batch_size = cfg.batch_size * (cfg.horizon+1) self._batch_size = cfg.batch_size * (cfg.horizon+1)
self._num_eps = 0 self._num_eps = 0