dppo/model/diffusion
Allen Z. Ren 1d04211666 v0.7 (#26)
* update from scratch configs

* update gym pretraining configs - use fewer epochs

* update robomimic pretraining configs - use fewer epochs

* allow trajectory plotting in eval agent

* add simple vit unet

* update avoid pretraining configs - use fewer epochs

* update furniture pretraining configs - use same amount of epochs as before

* add robomimic diffusion unet pretraining configs

* update robomimic finetuning configs - higher lr

* add vit unet checkpoint urls

* update pretraining and finetuning instructions as configs are updated
2024-11-20 15:56:23 -05:00
..
__init__.py release 2024-09-03 21:03:27 -04:00
diffusion_awr.py release 2024-09-03 21:03:27 -04:00
diffusion_dipo.py v0.5 to main (#10) 2024-10-07 16:35:13 -04:00
diffusion_dql.py v0.5 to main (#10) 2024-10-07 16:35:13 -04:00
diffusion_idql.py v0.5 to main (#10) 2024-10-07 16:35:13 -04:00
diffusion_ppo_exact.py v0.5 to main (#10) 2024-10-07 16:35:13 -04:00
diffusion_ppo.py v0.6 (#18) 2024-10-30 19:58:06 -04:00
diffusion_qsm.py v0.5 to main (#10) 2024-10-07 16:35:13 -04:00
diffusion_rwr.py add minor docs to diffusion classes and clean up some args 2024-09-17 16:26:25 -04:00
diffusion_vpg.py v0.6 (#18) 2024-10-30 19:58:06 -04:00
diffusion.py fix diffusion loss when predicting initial noise 2024-10-13 11:19:10 -04:00
eta.py squash commits 2024-09-11 21:09:17 -04:00
exact_likelihood.py squash commits 2024-09-11 21:09:17 -04:00
mlp_diffusion.py v0.5 to main (#10) 2024-10-07 16:35:13 -04:00
modules.py release 2024-09-03 21:03:27 -04:00
sampling.py squash commits 2024-09-11 21:09:17 -04:00
sde_lib.py release 2024-09-03 21:03:27 -04:00
unet.py v0.7 (#26) 2024-11-20 15:56:23 -05:00