Remove loss masking from diffusion policy (#135)

2024-05-06 07:27:01 +01:00
parent f5e76393eb
commit a8e245fb31
5 changed files with 10 additions and 1 deletions
--- a/lerobot/configs/policy/diffusion.yaml
+++ b/lerobot/configs/policy/diffusion.yaml
@@ -95,3 +95,6 @@ policy:

  # Inference
  num_inference_steps: 100
+
+  # Loss computation
+  do_mask_loss_for_padding: false