CLIP官方代码详解

醒了就刷牙

已于 2024-09-20 21:49:44 修改

阅读量271

点赞数 1

CC 4.0 BY-SA版权

分类专栏：论文文章标签：深度学习人工智能机器学习

于 2024-09-13 20:39:40 首次发布

本文链接：https://blog.youkuaiyun.com/buyaotutou/article/details/142219700

import os
import clip
import torch
from torchvision.datasets import CIFAR100

# Load the model
device = "cuda" if torch.cuda.is_available() else "cpu"
model, preprocess = clip.load('ViT-B/32', device)

# Download the dataset
cifar100 = CIFAR100(root=os.path.expanduser("~/.cache")