diff --git a/get_configs.py b/get_configs.py index 04eea88..de4fc43 100644 --- a/get_configs.py +++ b/get_configs.py @@ -3,14 +3,14 @@ def configs(model_name): args = { 'clip_initialized_model': - {"side_size": 224, - "crop_size": 256, + {"side_size": 256, + "crop_size": 224, "num_frames": 8, "mean": [0.48145466, 0.4578275, 0.40821073], "std": [0.26862954, 0.26130258, 0.27577711]}, 'frozen_model': - {"side_size": 224, - "crop_size": 256, + {"side_size": 256, + "crop_size": 224, "num_frames": 4, "mean": [0.485, 0.456, 0.406], "std": [0.229, 0.224, 0.225], }