From 661bda45ea422a94bb8f0cba069c9df7b5e9dc4e Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Sat, 2 Mar 2024 17:18:58 +0000 Subject: [PATCH] imagenet_norm: False --- lerobot/configs/policy/diffusion.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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