torch学习笔记
imperfect00
如果有梦想不去坚持实现,就是空想。
展开
-
luajit: not enough memory
运行代码错误:luajit: not enough memory查找原因貌似是因为 LuaJIT limit of 2GB,解决方法是安装我们可以安装安装LUA5.1,由于在安装torch的时候./install命令默认安装LuaJIT 2.1,我们可以在这一步选择安装LUA5.1,命令如下:TORCH_LUA_VERSION=LUA51 ./install.sh原创 2017-02-10 18:36:56 · 7270 阅读 · 0 评论 -
pytorch模型转ncnn,并移植android,ios
模型转换pytorch转onnximport torch.utils.datafrom torch.autograd import Variablefrom squeezenet import squeezenet1_2import onnxruntimeimport numpy as npfrom onnxruntime.datasets import get_exampleim...原创 2019-07-26 10:34:37 · 4068 阅读 · 2 评论 -
torch narrow用法
对于变量,有时候需要去变量的某一维度的某几个值,可以使用narrow函数实现,例如对于变量x,narrow函数输入为,`x.narrow`(*dimension*, *start*, *length*) → Tensor表示取变量x的第dimension维,的从索引start开始到,start+length范围的值.例如,我们取变量x的值为,x = torch.Tensor([[1, 2, 3],原创 2017-12-02 15:32:00 · 12905 阅读 · 0 评论 -
RuntimeError: invalid argument 4: out of range at pytorch/torch/lib/TH/generic/THTensor.c:439
使用from torchvision.utils import save_image保存图像时,出现错误:File “/usr/local/lib/python2.7/dist-packages/torchvision/utils.py”, line 51, in save_imagegrid = make_grid(tensor, nrow=nrow, padding=padding)File原创 2017-12-02 10:06:11 · 4680 阅读 · 5 评论 -
module 'nninit' not found:No LuaRocks module found for nninit
torch代码出现错误:module 'nninit' not found:No LuaRocks module found for nninit:这是因为缺少nninit模块,安装nninit即可,安装命令为:luarocks install nninit原创 2017-11-08 09:18:33 · 2383 阅读 · 1 评论 -
torch 变量更新
变量类型pytorch中有两种变量类型,一个是Tensor,一个是Variable。 Tensor: 就像ndarray一样,一维Tensor叫Vector,二维Tensor叫Matrix,三维及以上称为Tensor Variable:是Tensor的一个wrapper,不仅保存了值,而且保存了这个值的creator,需要BP的网络都是VariableVariableVariable包括dat原创 2017-10-29 11:35:42 · 1297 阅读 · 0 评论 -
ZeroBrane Studio命令行参数输入
终端运行lua代码为:th main.lua -netType vgg -save vgg_cifar10/ -S 0.0001如果采用ZeroBrane Studio进行调试,运行代码,则需要输入命令行参数. ZeroBrane Studio命令行参数输入方法为:点击Project,Command line parameters选项,输入需要输入的参数即可.原创 2017-08-29 19:48:41 · 1092 阅读 · 0 评论 -
ubuntu16.04安装torch
下载torch安装包: git clone https://github.com/torch/distro.git torch --recursive安装依赖库: cd torch/ sudo bash install-deps安装好后会提示:Torch7’s dependencies have been installed安装torch:sudo ./install.sh安装到最后会提示是否将原创 2017-08-29 16:38:36 · 6346 阅读 · 0 评论 -
pytorch安装
pytorch安装在http://pytorch.org/中点击对应的系统版本,即可以在网页上的run this command找到对应的命令行安装代码.我的电脑ubuntu 同时安装了python2.7和python3.5,安装pytorch for python3.5时,按照pytorch的安装教程终端输入命令行:sudo pip install http://download.pytorch.原创 2017-08-12 11:43:43 · 8108 阅读 · 0 评论 -
pytorch模型转ncnn,并移植android
模型转换pytorch转onnximport torch.utils.datafrom torch.autograd import Variablefrom squeezenet import squeezenet1_2import onnxruntimeimport numpy as npfrom onnxruntime.datasets import get_exampleim...原创 2019-10-10 11:54:42 · 4956 阅读 · 1 评论