小猴学IT

分类

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

/ 注册

全部文章（共90篇）

Python编程思想（21）：原来函数内部还可以定义函数（局部函数）

在前面的文章中所有的函数都是在全局范围内定义的，这些函数都称为全局函数。 Python还支持在函数体内定义函数，这种被放在函数体内定义的函数称为局部函数。在默认情况下，局部函数对外部是隐藏的，局部函数只能在函数体内调用。函数也可以返回局部函数，以便程序在其他作用域中使用局部数。示例代码：loca...

2021-04-29

0 765

python学习之argparse模块

argparse是python用于解析命令行参数和选项的标准模块。argparse模块的作用是用于解析命令行参数。我们很多时候，需要用到解析命令行参数的程序。二、使用步骤我们常常可以把argparse的使用简化成下面四个步骤 import argparse parser = argparse....

2021-04-22

0 607

torchvision.transforms用法介绍

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

2021-04-21

0 953

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

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

2021-04-21

0 634

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

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

2021-04-21

0 1534

图像卷积和池化操作后的特征图大小计算方法

2021-04-20

0 724

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

2021-04-20

0 502

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

2021-04-20

0 833

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

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

2021-04-20

0 4088

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 3229