Cadene ac90b9c3ee Fix diffusion (rm transpose), Add prefetch

2024-02-28 17:45:01 +00:00

..

datasets

Fix diffusion (rm transpose), Add prefetch

2024-02-28 17:45:01 +00:00

envs

Refactor configs to have env in seperate yaml + Fix training

2024-02-25 17:42:47 +00:00

policies

Fix diffusion (rm transpose), Add prefetch

2024-02-28 17:45:01 +00:00

__init__.py

Add common, refactor eval with eval_policy

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

logger.py

Refactor train, eval_policy, logger, Add diffusion.yaml (WIP)

2024-02-26 01:10:09 +00:00

utils.py

Add common, refactor eval with eval_policy

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