Pytorch
文章平均质量分 68
GZHermit
前进是痛苦的
停滞是舒适的
展开
-
PyTorch(1) torch.nn与torch.nn.functional之间的区别和联系
在阅读PyTorch官网的教程的时候,发现介绍如何利用Pytorch搭建一个神经网络的示例代码是这样写。import torchfrom torch.autograd import Variableimport torch.nn as nnimport torch.nn.functional as Fclass Net(nn.Module): def __init__(self):原创 2017-12-06 17:14:50 · 15039 阅读 · 7 评论 -
Pytorch(1) pytorch和tensorflow里面的maxpool
pytorch和tensorflow所含的maxpool,虽然名字相同,但是功能是不一样。之前在用pytorch复现darknet里面的yolo-v2时才发现这个问题。在yolov2的第六个maxpool的时候,kernel为2,stride为1,所以 按道理来说呢,输出的边size应该比输入的边size少1,但是yolo的设计是输入和输出的shape要相同。所以才发现了这个问题。不同之处在原创 2018-02-02 11:39:23 · 8347 阅读 · 1 评论 -
Pytorch(2) maxpool的ceil_mode
pytorch里面的maxpool,有一个属性叫ceil_mode,这个属性在api里面的解释是 ceil_mode: when True, will use ceil instead of floor to compute the output shape也就是说,在计算输出的shape的时候,如果ceil_mode的值为True,那么则用天花板模式,否则用地板模式。???...原创 2018-02-23 10:53:07 · 16967 阅读 · 1 评论