diff --git a/lerobot/configs/policy/diffusion.yaml b/lerobot/configs/policy/diffusion.yaml index d15a7c4e..6f18816a 100644 --- a/lerobot/configs/policy/diffusion.yaml +++ b/lerobot/configs/policy/diffusion.yaml @@ -80,7 +80,7 @@ obs_encoder: random_crop: True use_group_norm: True share_rgb_model: False - imagenet_norm: True + imagenet_norm: False # TODO(rcadene): was set to True rgb_model: #_target_: diffusion_policy.model.vision.model_getter.get_resnet