bridge-former
copied
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
19 lines
523 B
19 lines
523 B
|
|
|
|
def configs(model_name):
|
|
args = {
|
|
'clip_initialized_model':
|
|
{"side_size": 224,
|
|
"crop_size": 256,
|
|
"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,
|
|
"num_frames": 4,
|
|
"mean": [0.485, 0.456, 0.406],
|
|
"std": [0.229, 0.224, 0.225], }
|
|
}
|
|
return args[model_name]
|
|
|