![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch学习笔记
小白深度学习入门系列
hehehetanchaow
这个作者很懒,什么都没留下…
展开
-
pytorch学习笔记——编译错误记录
1.model.cuda(async=True) 改为:model.cuda(non_blocking=True) 。python版本python>=3.72.total_loss += loss_val.data[0]IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert...原创 2020-04-10 20:00:37 · 1114 阅读 · 0 评论 -
pytorch学习笔记——MNIST训练初学
#!/usr/bin/env python# coding: utf-8import torchimport torch.nn as nnimport torch.nn.functional as Fimport torch.optim as optimfrom torchvision import datasets, transformsfrom torch.autograd...原创 2020-04-10 10:56:29 · 266 阅读 · 1 评论 -
pytorch学习笔记——基础知识
1.读取图像数据Pytorch自定义读取数据的方式,主要用到两个类:torch.utils.data.Dataset和torch.utils.data.DataLoaderimg = cv2.imread(os.path.join(base_path, path).strip())img = torch.Tensor(img)import cv2import osimp...原创 2020-04-10 10:58:10 · 792 阅读 · 0 评论