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