imagenet_norm: False

2024-03-02 17:18:58 +00:00
parent 80785f8d0e
commit 661bda45ea
1 changed files with 1 additions and 1 deletions
--- 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