Simon Alibert 4631d36c05 Add get_safe_torch_device in policies

2024-03-20 18:38:55 +01:00

..

model

revert dp changes, make act and tdmpc batch friendly

2024-03-18 19:18:21 +00:00

diffusion_unet_image_policy.py

Integrate diffusion policy

2024-03-10 15:31:17 +01:00

policy.py

Add get_safe_torch_device in policies

2024-03-20 18:38:55 +01:00

pytorch_utils.py

Integrate pusht env from diffusion

2024-03-10 16:33:03 +01:00

replay_buffer.py

Integrate diffusion policy

2024-03-10 15:31:17 +01:00