手撕CLIP – Day2 – dataset 目录 手撕CLIP -- Day2 -- dataset Contrastive Language-Image Pre-Training (CLIP) 模型原理 CLIP 网络结构图 Dataset代码 - 利用Mnist数据集做分类 Dataset代码 Part1 库函数 Part2 初始化一个数据集的类 Part3 测试 参考 Contrastive Language-Image Pre-Training (CLIP) 模型原理 CLIP 网络结构图 CLIP 网络结构 Dataset代码 - 利用Mnist数据集做分类 Dataset代码 Part1 库函数 # 该模块主要为了实现CLIP的Mnist数据集的实现 ''' # Part1 引入相关的库函数 ''' import torch from torch.utils import data import torchvision