摘要: 微调流程 在源数据集(source dataset)上预训练一个网络模型,即源模型(source model) 创建一个新的网络模型,即目标模型(target model) 目标模型复制了源模型上除了输出层外的所有模型设计及参数。 我们假设这些模型参数包含了源数据集上学习到的知识,且这些知识同样适用 阅读全文
posted @ 2022-04-06 22:18 ArdenWang 阅读(136) 评论(0) 推荐(0) 编辑
摘要: 修改模型 import torchvision.models as models net = models.resnet50() # 查看模型定义 print(net) # output ResNet( (conv1): Conv2d(3, 64, kernel_size=(7, 7), strid 阅读全文
posted @ 2022-04-06 20:39 ArdenWang 阅读(40) 评论(0) 推荐(0) 编辑
摘要: 在线下载 from PIL import Image import torch import torchvision from torch.utils.data.dataset import Dataset import torchvision.transforms as transforms # 阅读全文
posted @ 2022-04-06 10:11 ArdenWang 阅读(68) 评论(0) 推荐(0) 编辑
摘要: Tensor dtype: 张量的数据类型,如torch.FloatTensor, torch.cuda.FloatTensor shape: 张量形状 device: 张量所在设备 requires_grad: 指示是否需要梯度 grad: data的梯度 grad_fn: 创建Tensor的Fu 阅读全文
posted @ 2022-04-06 09:09 ArdenWang 阅读(88) 评论(0) 推荐(0) 编辑