allenzren
e7f73dffc1
update batch size in D3IL so it works with the new form of gradient update
2024-12-24 02:06:17 -05:00
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
allenzren
0bdae945e9
use default epoch_start_ema=20 and update_ema_freq=10
2024-11-07 10:55:16 -05:00
allenzren
c0921a1fb5
remove update_ema_freq
2024-11-06 21:01:15 -05:00
Allen Z. Ren
e1ef4ca1cf
More frequent EMA update ( #20 )
...
* move ema update within pretraining epoch
* update pretraining ema configs
* add lift and can epoch 8000 checkpoint url
* add note about EMA issue in pretraining instruction
2024-11-06 20:42:31 -05:00
Allen Z. Ren
e0842e71dc
v0.5 to main ( #10 )
...
* v0.5 (#9 )
* update idql configs
* update awr configs
* update dipo configs
* update qsm configs
* update dqm configs
* update project version to 0.5.0
2024-10-07 16:35:13 -04:00
allenzren
2ddf63b8f5
squash commits
2024-09-11 21:09:17 -04:00
allenzren
8ce0aa1485
simplify pre-training dataset, use npz
2024-09-08 17:52:16 -04:00
allenzren
8293b0936b
release
2024-09-03 21:03:27 -04:00
allenzren