pytorch学习笔记
小白深度学习入门系列
hehehetanchaow
这个作者很懒,什么都没留下…
展开
-
pytorch学习笔记——编译错误记录
1.model.cuda(async=True) 改为:model.cuda(non_blocking=True) 。python版本python>=3.7 2. 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 · 1106 阅读 · 0 评论 -
pytorch学习笔记——MNIST训练初学
#!/usr/bin/env python # coding: utf-8 import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torchvision import datasets, transforms from torch.autograd...原创 2020-04-10 10:56:29 · 250 阅读 · 1 评论 -
pytorch学习笔记——基础知识
1.读取图像数据 Pytorch自定义读取数据的方式,主要用到两个类:torch.utils.data.Dataset和torch.utils.data.DataLoader img = cv2.imread(os.path.join(base_path, path).strip()) img = torch.Tensor(img) import cv2 import os imp...原创 2020-04-10 10:58:10 · 780 阅读 · 0 评论