Cadene aed02dc7c6 Add multithreading for video generation, Speed policy sampling

2024-02-24 18:18:39 +00:00

..

datasets

Fix done in pusht, Fix --time in sbatch

2024-02-22 17:58:26 +00:00

envs

fix simxarm factory

2024-02-22 13:04:24 +00:00

__init__.py

Add common, refactor eval with eval_policy

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

logger.py

remove comments

2024-02-22 12:15:14 +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 multithreading for video generation, Speed policy sampling

2024-02-24 18:18:39 +00:00

utils.py

Add common, refactor eval with eval_policy

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