- 博客(8)
- 收藏
- 关注
原创 Pytorch学习笔记:根据optimizer_type选择优化器
文章目录1.根据optimizer_type选择优化器2.pytorch中冻结某些层设置1.根据optimizer_type选择优化器代码如下:#---------------------------------------## 根据optimizer_type选择优化器#---------------------------------------#pg0, pg1, pg2 = [], [], [] for k, v in model.named_modules(): ..
2022-03-29 16:48:29 2288
原创 Pytorch学习笔记:最简单的推理网络组成(from tudui)
文章目录前言1.引入库2.读取数据3.数据处理(totensor,resize)4.重载网络模型和训练好的参数5.输入图片给模型前言摘要:最简单的推理网络组成(from tudui)代码如下(示例):1.引入库import torchimport torchvisionfrom PIL import Imagefrom torch import nn2.读取数据image_path = "../imgs/airplane.png"image = Image.open(ima..
2022-03-11 18:28:20 2669
原创 Pytorch学习笔记:最简单的训练网络组成(from tudui)
文章目录前言一、使用步骤1.引入库2.定义训练的设备3.载入数据集4.DataLoader 来加载数据集5.创建网络模型并载入6.定义损失函数和优化器6.训练模型6.测试集验证前言摘要:一、使用步骤代码如下:1.引入库import torchimport torchvisionfrom torch.utils.tensorboard import SummaryWriter2.定义训练的设备device = torch.device("cuda")3.载入数据集# 载入..
2022-03-11 18:19:43 709
原创 调试报错-调用tensorboard 报错
1.调用tensorboard 报错:tensorboard --logdir=./logsFatal error in launcher: Unable to create process using2.修改如下python -m tensorboard.main --logdir=./logs
2022-02-20 16:35:59 1271
原创 Pytorch学习笔记:model.train()和model.eval()
文章目录前言1.model.train()2.model.eval()总结前言训练时的model.train()和测试时的model.eval()分别做了什么?1.model.train()源码如下:def train(self: T, mode: bool = True) -> T: r"""Sets the module in training mode. This has any effect only on certain modules. See doc..
2022-02-11 18:22:27 1575
原创 Pytorch学习笔记:自定义数据集
文章目录前言1.torch.utils.data.Dataset介绍2.实例a.准备数据集b.复写 Datasetc.DataLoader加载总结前言初学Pytorch时,数据集直接使用torchvision.datasets调用,然和直接使用torch.untils.data.DataLoader加载。在实际项目中,我们怎么自定义数据集呢?1.torch.utils.data.Dataset介绍`torch.utils.data.Dataset` 是一个抽象类,用户想要加载自定义的数..
2022-02-11 15:26:01 1333
原创 Pytorch学习笔记:加载预训练模型
文章目录前言摘要:一、几种不同情况1.加载模型2.修改加载模型3.加载预训练模型总结前言torchvision是官方经常需要的包,包括:torchvision.datasets:预定义的训练集(比如MNIST、CIFAR10等);torchvision.models:包含预定义好的经典网络结构(比如AlexNet、VGG、ResNet等),torchvision.transforms:数据增强的方法模型地址:https://github.com/pytorch/vision/tree/ma..
2022-02-11 13:39:51 4128
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人