lucidrains d5b70e2b86 allow for adding an RNN before time attention, but need to handle caching still

2025-11-10 11:42:20 -08: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

allow for adding an RNN before time attention, but need to handle caching still

2025-11-10 11:42:20 -08:00

mocks.py

redo so that max timesteps is treated as truncation at the last timestep, then allow for accepting the truncation signal from the environment and reuse same logic

2025-10-28 08:04:48 -07:00

trainers.py

able to move the experience to cpu easily, and auto matically move it to the device of the dynamics world model when learning from it

2025-11-09 16:16:13 +00:00