Alexander Soare 342f429f1c Add test to make sure policy dataclass configs match yaml configs (#292)		2024-06-26 09:09:40 +01:00
..
act	Bug fix: missing attention mask in VAE encoder in ACT policy (#279)	2024-06-19 12:07:21 +01:00
diffusion	fix Unet global_cond_dim to use state dim, not action dim (#278)	2024-06-17 15:17:28 +01:00
tdmpc	Improves Type Annotations (#252)	2024-06-10 19:09:48 +01:00
vqbet	Add VQ-BeT (#166)	2024-06-26 08:55:02 +01:00
factory.py	Add test to make sure policy dataclass configs match yaml configs (#292)	2024-06-26 09:09:40 +01:00
normalize.py	Fix bug in normalize to avoid divide by zero (#239)	2024-06-04 12:21:28 +02:00
policy_protocol.py	Improves Type Annotations (#252)	2024-06-10 19:09:48 +01:00
utils.py	Make policies compatible with other/multiple image keys (#149)	2024-05-16 13:51:53 +01:00