喜乐有分享

分类

题解(2)

/ 注册

全部文章（共4篇）

torchvision.transforms记录

PyTorch框架中有一个非常重要且好用的包：torchvision，该包主要由3个子包组成，分别是： torchvision.datasets torchvision.models torchvision.transforms 这3个子包的具体介绍可以参考官网：http://...

Pytorch

2019-06-20

1 1421

pytorch_optim_学习率调整策略

PyTorch学习率调整策略通过torch.optim.lr_scheduler接口实现。PyTorch提供的学习率调整策略分为三大类，分别是 a. 有序调整：等间隔调整(Step)，按需调整学习率(MultiStep)，指数衰减调整(Exponential)和余弦退火CosineAnneali...

Pytorch

2019-06-22

0 934

pytorch 优化器(optim)不同参数组，不同学习率设置

optim 的基本使用 for do: 1. 计算loss 2. 清空梯度 3. 反传梯度 4. 更新参数 cifiron = nn.MSELoss() optimiter = torch.optim.SGD(net.parameters(),lr=0.01,momentum=0.9) for...

Pytorch

2019-07-01

0 2853

pytorch 实现roi pool

# -*- coding:UTF-8 -*- import time import torch import torch.nn.functional as F from torch.autograd import Variable def roi_pooling(input, rois, si...

Pytorch

2019-08-07

0 584