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")
CLIP官方代码详解
于 2024-09-13 20:39:40 首次发布

最低0.47元/天 解锁文章
6487

被折叠的 条评论
为什么被折叠?



