Cadene ece89730e6 Add pusht dataset (TODO verify reward is aligned), Refactor visualize_dataset, Add video_dir, fps, state_dim, action_dim to config (Training works)

2024-02-21 00:49:40 +00:00

..

datasets

Add pusht dataset (TODO verify reward is aligned), Refactor visualize_dataset, Add video_dir, fps, state_dim, action_dim to config (Training works)

2024-02-21 00:49:40 +00:00

envs

Add Prod transform, Add test_factory

2024-02-20 14:22:16 +00:00

__init__.py

Add common, refactor eval with eval_policy

2024-01-31 13:48:12 +00:00

logger.py

Eval reproduced! Train running (but not reproduced)

2024-02-10 15:46:24 +00:00

tdmpc_helper.py

Add pusht dataset (TODO verify reward is aligned), Refactor visualize_dataset, Add video_dir, fps, state_dim, action_dim to config (Training works)

2024-02-21 00:49:40 +00:00

tdmpc.py

Add pusht dataset (TODO verify reward is aligned), Refactor visualize_dataset, Add video_dir, fps, state_dim, action_dim to config (Training works)

2024-02-21 00:49:40 +00:00

utils.py

Add common, refactor eval with eval_policy

2024-01-31 13:48:12 +00:00