diff --git a/frozen_in_time.py b/frozen_in_time.py index 9978b7a..209b9fb 100644 --- a/frozen_in_time.py +++ b/frozen_in_time.py @@ -73,8 +73,8 @@ class FrozenInTime(NNOperator): self.tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', TOKENIZERS_PARALLELISM=False) self.transform_cfgs = get_configs( - side_size=224, - crop_size=256, + side_size=256, + crop_size=224, num_frames=self.num_frames, mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225],