Alexander Soare 88347965c2 revert dp changes, make act and tdmpc batch friendly

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

..

conditional_unet1d.py

Integrate diffusion policy

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

conv1d_components.py

Integrate diffusion policy

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

crop_randomizer.py

Integrate diffusion policy

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

dict_of_tensor_mixin.py

Integrate diffusion policy

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

ema_model.py

Integrate pusht env from diffusion

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

lr_scheduler.py

Integrate diffusion policy

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

mask_generator.py

Integrate diffusion policy

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

module_attr_mixin.py

Integrate diffusion policy

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

multi_image_obs_encoder.py

revert dp changes, make act and tdmpc batch friendly

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

normalizer.py

Integrate diffusion policy

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

positional_embedding.py

Integrate diffusion policy

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

tensor_utils.py

Integrate diffusion policy

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