# Copyright (c) 2022, salesforce.com, inc. | |
# All rights reserved. | |
# SPDX-License-Identifier: BSD-3-Clause | |
# For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause | |
model: | |
arch: clip | |
model_type: ViT-L-14 | |
# ['RN50', | |
# 'RN50-quickgelu', | |
# 'RN50x4', | |
# 'RN50x16', | |
# 'RN101', | |
# 'RN101-quickgelu', | |
# 'timm-efficientnetv2_rw_s', | |
# 'timm-resnet50d', | |
# 'timm-resnetaa50d', | |
# 'timm-resnetblur50', | |
# 'timm-swin_base_patch4_window7_224', | |
# 'timm-vit_base_patch16_224', | |
# 'timm-vit_base_patch32_224', | |
# 'timm-vit_small_patch16_224', | |
# 'ViT-B-16', | |
# 'ViT-B-16-plus', | |
# 'ViT-B-16-plus-240', | |
# 'ViT-B-32', | |
# 'ViT-B-32-plus-256', | |
# 'ViT-B-32-quickgelu', | |
# 'ViT-g-14', | |
# 'ViT-H-14', | |
# 'ViT-H-16', | |
# 'ViT-L-14', | |
# 'ViT-L-14-280', | |
# 'ViT-L-14-336', | |
# 'ViT-L-16', | |
# 'ViT-L-16-320'] | |
pretrained: openai | |
# "openai" | |
# following not available for all models | |
# "yfcc15m" | |
# "cc12m" | |
# "laion400m_e31" | |
# "laion400m_e32" | |
# "laion400m_avg" | |
preprocess: | |
vis_processor: | |
eval: | |
name: "clip_image_eval" | |
image_size: 224 | |