在 nvidia nsight 上调试新的 caffe 层

Question

我想在 nsight 上为 caffe(.cpp, .cu, .hpp) 调试一个新的层写入，我想知道通常情况下，你会 wrtie对该层进行测试并调试其中包含 main 的 test.cpp 并调用新层？我应该写一些类似于 caffe 中的图层测试文件的东西吗？

Answer 1

你绝对应该为你的图层编写一个测试，测试 forward 和梯度计算（使用自动数值梯度测试工具）。
一旦你完成了这个测试，就可以直接确定你的层是否正常工作，如果没有，那么在没有 运行 整个框架的情况下调试它会更容易。