Simon Alibert 4631d36c05 Add get_safe_torch_device in policies

2024-03-20 18:38:55 +01: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

Add get_safe_torch_device in policies

2024-03-20 18:38:55 +01: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