Sue Hyun Park 8bbc14ebab Fix: handle _action_masks buffer in single-task scenarios (#67)

2025-05-19 20:12:12 -07:00

..

common

Fix: handle _action_masks buffer in single-task scenarios (#67)

2025-05-19 20:12:12 -07:00

envs

full support for episodic rl

2025-04-15 15:55:05 -07:00

trainer

full support for episodic rl

2025-04-15 15:55:05 -07:00

__init__.py

first commit

2023-10-25 18:26:00 -07:00

config.yaml

update readme + clean up

2025-04-15 16:32:15 -07:00

evaluate.py

partial fix to loading checkpoints

2025-01-21 00:10:53 -08:00

tdmpc2.py

only instantiate termination pred head if episodic=true

2025-05-02 16:51:24 -07:00

train.py

full support for episodic rl

2025-04-15 15:55:05 -07:00