Alexander Soare 87fcc536f9 wip - still need to verify full training run

2024-03-11 18:45:21 +00:00

..

model

wip - still need to verify full training run

2024-03-11 18:45:21 +00:00

diffusion_unet_image_policy.py

Integrate diffusion policy

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

policy.py

early training loss as expected

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