小猴学IT

pytorch

c++(2) Java(1) Linux(1) mmdetection(1) MySQL数据库(3) opencv(9) paddlepaddle(2) python(4) ZHY项目仓库(2) 学习记录(17) 数据分析及可视化(1) 机器学习(7) 爬虫(1) 目标检测(1) 程序报错解决方案汇总(5) 自然语言处理(2) 题解(16)

/ 注册

全部文章 / pytorch （共19篇）

pth转onnx模型错误.Error: ‘net‘ object has no attribute ‘co

使用torch.load_state_dict()将pth转onnx模型出现的问题 model = ResNet34(num_classes=20) model.load_state_dict(torch.load(model_path, map_location=device),strict=Fa...

2021-09-08

0 2099

torch.utils.data.DataLoader（）详解

使用方法以及要点一、不用sampler # 训练数据集的加载器，自动将数据分割成batch，顺序随机打乱 train_loader = torch.utils.data.DataLoader(dataset=train_dataset, ...

2021-05-04

0 2125

Tensor转为PILimage格式

from PIL import Image import matplotlib.pyplot as plt from torchvision import transforms to_img = transforms.ToPILImage() img = to_img(imgTensor*0.2...

2021-05-03

0 415

torchvision.transforms用法介绍

pytorch源码解读之torchvision.transforms PyTorch框架中有一个非常重要且好用的包：torchvision，该包主要由3个子包组成，分别是：torchvision.datasets、torchvision.models、torchvision.transforms。这...

2021-04-21

0 767

pytorch进行CIFAR分类（数据加载和处理、网络结构、定义损失函数和优化器、训练测试）

Pytorch打怪路系列Pytorch打怪路（一）pytorch进行CIFAR-10分类（1）CIFAR-10数据加载和处理链接说明Pytorch打怪路（一）pytorch进行CIFAR-10分类（2）定义卷积神经网络Pytorch打怪路（一）pytorch进行CIFAR-10分类（3）定义损失函数...

2021-04-21

0 479

PyTorch加载预训练模型进行微调

在此示例中，我们从中加载了经过预训练的resnet18模型torchvision。我们创建一个随机数据张量以表示具有3个通道，高度和宽度为64的单个图像，并将其对应的label值初始化为一些随机值。 import torch, torchvision model = torchvision.mode...

2021-04-21

0 1315

Pytorch 为什么每一轮batch需要设置optimizer.zero_grad

2021-04-20

0 376

PyTorch系列 | _, predicted = torch.max(outputs.data, 1)的理解

2021-04-20

0 675

pytorch的nn.CrossEntropyLoss()函数使用方法

nn.CrossEntropyLoss()函数计算交叉熵损失用法： # output是网络的输出，size=[batch_size, class] #如网络的batch size为128，数据分为10类，则size=[128, 10] # target是数据的真实标签，是标量，size=[bat...

2021-04-20

0 3613

Pytorch模型数据的gpu和cpu：model.to(device), model.cuda()

背景介绍我们在使用Pytorch训练时，模型和数据有可能加载在不同的设备上（gpu和cpu），在算梯度或者loss的时候，报错信息类似如下： RuntimeError: Function AddBackward0 returned an invalid gradient at index 1 - e...

2021-04-06

0 2669