Pytorch学习
萌笙
初窥门径CVer
展开
-
Pytorch03: 使用预训练模型训练 Kaggle: Dogs vs Cats
Pytorch使用预训练模型训练 Kaggle: Dogs vs Cats1、Pytorch加载数据集 pytorch 的torchvision中继承了常用的数据集,可以直接在torchvision.datasets中调用,当我们想要加载自己的数据集来训练模型时,需要借助torchvision.datasets.ImageFolder这个类,在当前的Pytorch版本中, ImageFolder继承自DatasetFolder官方源码 我们可以直接使用ImageFold原创 2020-10-07 16:08:54 · 703 阅读 · 0 评论 -
Pytorch笔记02:构建神经网络—以CIFA-10数据集分类为例
Pytorch笔记02:构建神经网络—以CIFA-10数据集分类为例1、pytorch加载内置数据集在Pytorch的torchvision中集成了很多常用的数据集,例如MNIST、CIFA-10、COCO等,使用的方法就是在torchvision.dataset内直接使用,如torchvision.datasets.cifa10(),一般加载数据集有这样几个参数:root(str): 表示数据集文件的路径train(bool): 如果是True表示从train set中创建,否则从test se原创 2020-10-07 16:04:57 · 610 阅读 · 0 评论 -
Pytorch笔记01:随机数种子的设置 VS zero_grad()
Pytorch 笔记01Pytorch中随机数种子的设置 为了模型结果可复现,常常需要为程序设置随机数种子(seed),在使用Pytorch进行模型训练时,有以下几个部分可以设置随机种子。1、cudnnfrom torch.backends import cudnncudnn.benchmark = False cudnn.deterministic = True2、Pytorchimport torchseed = 121torch.manual_seed(seed)原创 2020-10-07 15:59:36 · 459 阅读 · 0 评论