Cadene e3643d6146 Wandb works, One output dir

2024-02-22 12:14:12 +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

Wandb works, One output dir

2024-02-22 12:14:12 +00:00

__init__.py

Add common, refactor eval with eval_policy

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

logger.py

Wandb works, One output dir

2024-02-22 12:14:12 +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

Wandb works, One output dir

2024-02-22 12:14:12 +00:00

utils.py

Add common, refactor eval with eval_policy

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