Alexander Soare 2a01487494 early training loss as expected

2024-03-11 13:34:04 +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

early training loss as expected

2024-03-11 13:34:04 +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