lucidrains 995b1f64e5 handle environments that return a terminate flag, also make sure episode lens are logged in vectorized env

2025-10-27 10:14:28 -07:00

..

__init__.py

first pass through gathering experience with a mock env for online rl

2025-10-22 08:32:46 -07:00

dreamer4.py

handle environments that return a terminate flag, also make sure episode lens are logged in vectorized env

2025-10-27 10:14:28 -07:00

mocks.py

handle environments that return a terminate flag, also make sure episode lens are logged in vectorized env

2025-10-27 10:14:28 -07:00

trainers.py

redo so lens is always the episode length, including the bootstrap value timestep, and use is_truncated to mask out the bootstrap node from being learned on

2025-10-27 08:06:21 -07:00