在使用PyTorch的时候,经常遇到nn.Sequential和nn.ModuleList,今天将这两个模块认真区分了一下,总结如下。PyTorch版本为1.0.0。本文也会随着本人逐渐深入Torch和有新的体会时,会进…
标签:pytorch
PyTorch转TensorRT流程
1.安装TensorRT 很多教程,可以谷歌 2.安装onnx sudo apt-get install protobuf-compiler libprotoc-dev pip install onnx 3.安装 onn…
浅谈 PyTorch 中的 tensor 及使用
这篇文章主要是围绕 PyTorch 中的 tensor 展开的,讨论了张量的求导机制,在不同设备之间的转换,神经网络中权重的更新等内容。面向的读者是使用过 PyTorch 一段时间的用户。本文中的代码例子基于 Pytho…
神经网络调参,提升准确率
—— 如果只是调cnn网络的参数,可以看这篇文章 http://cs231n.github.io/neural-networks-2/#init —— 当初步搭建好网络,模…
pytorch之 activation funcion
1 import torch 2 import torch.nn.functional as F 3 from torch.autograd import Variable 4 import matplotlib.pyp…
Pytorch 基础
Pytorch 1.0.0 学习笔记: Pytorch 的学习可以参考:Welcome to PyTorch Tutorials Pytorch 是什么? 快速上手 Pytorch! Tensors(张量) from _…
Pytorch--Dropout笔记
dropout常常用于抑制过拟合,pytorch也提供了很方便的函数。但是经常不知道dropout的参数p是什么意思。在TensorFlow中p叫做keep_prob,就一直以为pytorch中的p应该就是保留节点数的比…
PyTorch 初体验(一个优雅的框架)
pytorch是啥呢?其实pytorch是一个python优先的深度学习框架,是一个和tensorflow,Caffe,MXnet一样,非常底层的框架,它的前身是torch,主要的语言接口是Lua,在如今github上前…
pytorch预训练模型的下载地址以及解决下载速度慢的方法
https://github.com/pytorch/vision/tree/master/torchvision/models 几乎所有的常用预训练模型都在这里面 总结下各种模型的下载地址: 1 Resnet: 2 3…
PyTorch的量化
先占坑,待续。 The quantized folder holds the implementation of the low-level quantized kernel.The kernels are regist…
PyTorch源码浅析(五)
我前几天在物理所为了讲Julia的多重派发写了一个C的实现,其实在PyTorch里也是利用这个机制来实现运行时(runtime)根据底层类型调用不同的SIMD加速的。关于中科院物理所的数值俱乐部:http://num.v…
Image Captioning代码复现
Image caption generation: https://github.com/eladhoffer/captionGen Simple encoder-decoder image captioning: ht…