def configs(model_name): args = { 'clip_initialized_model': {"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": 256, "crop_size": 224, "num_frames": 4, "mean": [0.485, 0.456, 0.406], "std": [0.229, 0.224, 0.225], } } return args[model_name]