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

update pinned torchrl version

Browse Source
This commit is contained in:
Nicklas Hansen
2024-03-27 12:51:40 -07:00
parent d51feb0e9f
commit 3789fcd5b8
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docker/environment.yaml
View File

@@ -39,9 +39,9 @@ dependencies:
- protobuf==4.25.2 - protobuf==4.25.2
- pillow==10.2.0 - pillow==10.2.0
- pyquaternion==0.9.9 - pyquaternion==0.9.9
- tensordict-nightly==2024.1.10 - tensordict-nightly==2024.3.26
- termcolor==2.4.0 - termcolor==2.4.0
- torchrl-nightly==2024.1.10 - torchrl-nightly==2024.3.26
- transforms3d==0.4.1 - transforms3d==0.4.1
- trimesh==4.0.9 - trimesh==4.0.9
- tqdm==4.66.1 - tqdm==4.66.1

1
tdmpc2/common/buffer.py
View File

@@ -19,6 +19,7 @@ class Buffer():
end_key=None, end_key=None,
traj_key='episode', traj_key='episode',
truncated_key=None, truncated_key=None,
strict_length=True,
) )
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