diff --git a/diffusion_policy/model/vision/multi_image_obs_encoder.py b/diffusion_policy/model/vision/multi_image_obs_encoder.py index de6aa658..c4f6755e 100644 --- a/diffusion_policy/model/vision/multi_image_obs_encoder.py +++ b/diffusion_policy/model/vision/multi_image_obs_encoder.py @@ -98,7 +98,7 @@ def __init__(self, pos_enc=False ) else: - this_normalizer = torchvision.transforms.CenterCrop( + this_randomizer = torchvision.transforms.CenterCrop( size=(h,w) ) # configure normalizer