自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(8)
  • 收藏
  • 关注

原创 conda常用命令

conda常用命令

2022-11-04 16:56:55 3660 1

原创 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关注的人

提示
确定要删除当前文章?
取消 删除