Pytorch源码学习
mathlxj
复旦大学数院博士毕业,蚂蚁打工人,神经网络方向,目前涉猎计算机视觉、机器学习、推荐系统、TensorRT、自动驾驶、联邦学习、消费信贷、量化等领域。
展开
-
pytorch转onnx, onnx 12 中没有hardswish opt
转onnx不支持hardswish激活函数的解决方法原创 2022-04-09 16:28:04 · 2231 阅读 · 1 评论 -
pytorch+FasterRCNN在Jetson Xavier NX上的测速
测速结果为原创,转载需附带链接。基本配置:Jetpack 4.4 DP, Pytorch 1.2 for DP原创 2020-11-17 10:01:53 · 1546 阅读 · 0 评论 -
pytorch+FasterRCNN在Jetson Xavier上的测速
测速结果为原创,转载需附带本文链接。基本配置:ResNet50+FPN,输出尺寸600x1000原创 2020-11-17 09:55:23 · 1147 阅读 · 0 评论 -
Pytorch学习导航页
本博客中与pytorch学习相关的博文,方便查看。1.Pytorch源码学习之一: torchvision.models.alexnet2.Pytorch源码学习之二:torchvision.models.vgg3.Pytorch源码学习之三:torchvision.models.resnet4.Pytorch源码学习之四:torchvision.models.squeezenet5.Pytorch源码学习之五:torchvision.models.densenet6.Pytorch源码学习之六原创 2020-06-14 16:11:06 · 292 阅读 · 0 评论 -
Pytorch源码学习之七:torchvision.models.googlenet
0.基本知识torchvision.models.googlenet源码地址GoogLeNet论文地址Inception模块注意,代码中第三个模块采取的3x3卷积而非5x5卷积,不知为何。-GoogLeNet网络结构GoogLeNet网络结构可视化这里注意两个辅助的输出。一、源码import torchimport torch.nn as nnimport torc...原创 2020-03-27 11:16:24 · 2274 阅读 · 0 评论 -
Pytorch源码学习之六:torchvision.models.inception_v3
0. 参考和简介torchvision.models.inception_v3源码实现的论文地址部分网络结构:InceptionAInceptionCInceptionE一、源码#网络模型:https://arxiv.org/abs/1512.00567from collections import namedtupleimport torchimport torc...原创 2020-03-26 13:21:04 · 4626 阅读 · 2 评论 -
Pytorch源码学习之五:torchvision.models.densenet
0. 基本知识DenseNet论文地址DenseNet加强了每个Dense Block内部的连接,每层输出与之前所有层进行concat连接,使用三个Dense Block的网络示意图如下:每个Block之间使用Transition(BN+ReLU+Conv+AvePool),最后使用全连接层作为分类器.torchvision实现的DenseNet网络结构同论文,如下:1.torchv...原创 2020-03-25 11:35:03 · 3084 阅读 · 1 评论 -
Pytorch源码学习之四:torchvision.models.squeezenet
0.介绍Squeezenet网址torchvision.model.squeeze官方文档主要思想:堆叠Fire模块,每个Fire模块,分别采用1x1和3x3两个分支,最后做拼;,每个Fire的尺寸不变,channel数不变或增加;每个stage的Fire模块之间用nn.MaxPool2d进行下采样;使用卷积层代替FC层,channel数为类别数1.源码import torchimpo...原创 2020-03-24 11:44:59 · 1612 阅读 · 0 评论 -
Pytorch源码学习之三:torchvision.models.resnet
一、网络结构1.BasicBlock2.BootleNeck和group convolution3.Dilated Convolution with a 3 x 3 kernel and dilation rate 2二、torchvision源码源码地址为:torchvision.models.resnetresnetxt50_32x4d和resnext101_32x8d见论文A...原创 2020-03-20 14:14:24 · 7969 阅读 · 3 评论 -
Pytorch源码学习之二:torchvision.models.vgg
0. VGG的网络结构一、torchvision源码这种通过配置文件一次性搭建相似网络的结构的方法十分值得学习和模仿.这也是相对于AlexNet的实现过程不同之处.我对其做了一丁点修改,源码网址可见torchvision.models.vgg源码网址'''VGG的torchvison实现重写,'''import torchimport torch.nn as nntry: ...原创 2020-03-18 19:49:24 · 3920 阅读 · 0 评论 -
Pytorch源码学习之一: torchvision.models.alexnet
Pytorch源码学习之一 torchvision.models.alexnettorch中一行代码导入的alexnet配置如下:64@11x11 stride=4 + ReLU + Maxpooling 3x3 stride=2192@5x5 stride=1 + ReLU + MaxPooling 3x3 stride=2384@3x3 stride=1 + ReLU256@3...原创 2020-03-17 16:39:01 · 2785 阅读 · 0 评论