lucidrains fe79bfa951 optionally keep track of returns statistics and normalize with them before advantage

2025-10-27 09:02:08 -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

optionally keep track of returns statistics and normalize with them before advantage

2025-10-27 09:02:08 -07:00

mocks.py

handle vectorized env

2025-10-22 11:19:44 -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