Commit Graph

15 Commits

Author SHA1 Message Date
allenzren
dd14c5887c set deterministic=True when sampling in diffusion evaluation 2024-09-26 01:15:10 -04:00
allenzren
4962bbce38 rename train.py to run.py 2024-09-17 16:33:53 -04:00
allenzren
c9f24ba0c3 add evaluation agents and some example configs 2024-09-17 16:32:45 -04:00
allenzren
bc52beca1e add minor docs to diffusion classes and clean up some args 2024-09-17 16:26:25 -04:00
allenzren
ef5b14f820 fix observation history indexing in the dataset 2024-09-17 12:43:15 -04:00
allenzren
1aaa6c2302 support varying img size 2024-09-16 17:55:31 -04:00
allenzren
64595baca9 more intuitive handling of done in GAE 2024-09-13 16:29:56 -04:00
allenzren
f5a8da5719 typo 2024-09-11 21:50:02 -04:00
allenzren
f13eb203e1 allow history observation 2024-09-11 21:44:47 -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
447c8dfd02 update instruction on reducing cpu threads 2024-09-08 13:43:52 -04:00
allenzren
a658353eb7 update compute instruction 2024-09-04 13:19:39 -04:00
allenzren
771240b7a6 fix repo link 2024-09-03 21:05:26 -04:00
allenzren
8293b0936b release 2024-09-03 21:03:27 -04:00