torch 修改层(未完)

1、如果只是对nn下的层进行修改,例如修改xx.lua文件,不需要重新编译整个torch只需要

cd torch/extra/nn/
luarocks make rocks/nn-scm-1.rockspec

2、加入一个只涉及lua操作的新层,从nn继承,修改/extra/nn/init.lua文件,假设加入的是xx.lua问价,那么init.lua里面要加上require(‘nn.xx’)

cd torch/extra/nn/
luarocks make rocks/nn-scm-1.rockspec

3、如果修改了c文件中函数的定义,记得修改对应目录下的torch/extra/nn/lib/THNN/generic/THNN.h中的函数定义!!!然后

cd torch/extra/nn/
luarocks make rocks/nn-scm-1.rockspec

即可

4、修改了cu文件

cd /torch/extra/cunn
luarocks make rocks/cunn-scm-1.rockspec

猜你喜欢

转载自blog.csdn.net/u013548568/article/details/79688306