Alexander Soare ba91976944 wip: still needs batch logic for act and tdmp

2024-03-14 15:24:10 +00:00

..

backbone.py

Add Aloha env and ACT policy

2024-03-12 10:27:48 +00:00

detr_vae.py

Add Aloha env and ACT policy

2024-03-12 10:27:48 +00:00

policy.py

wip: still needs batch logic for act and tdmp

2024-03-14 15:24:10 +00:00

position_encoding.py

Add Aloha env and ACT policy

2024-03-12 10:27:48 +00:00

transformer.py

Add Aloha env and ACT policy

2024-03-12 10:27:48 +00:00

utils.py

Add Aloha env and ACT policy

2024-03-12 10:27:48 +00:00