![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
CNN
文章平均质量分 54
biggerbugger
这个作者很懒,什么都没留下…
展开
-
CNN---pytorch---MNIST---入门(简单)分类任务
分类任务MNIST一级目录定义模型训练网络 一级目录 # 导包 import torch import torchvision import torchvision.transforms as transforms device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") print(device) # cuda:0 # 下载数据、读取数据 transform = torchvision.transforms.Com原创 2021-03-06 15:42:51 · 322 阅读 · 0 评论 -
CIFAR10分类任务---pytorch---CNN---入门(简单)分类任务
本实验来自Pytorch官方教程中文版 CIFAR10分类任务 # 导包 import torch import torchvision import torchvision.transforms as transforms # device = torch.device("cuda:0") # 指定模型训练所在 GPU device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") print(device) # c原创 2021-03-03 16:18:21 · 1758 阅读 · 1 评论