在nvidia上展示新的caffe图层

Question

我想在nsight上调试caffe（.cpp，.cu，.hpp）的新图层，我想知道，你是否会参加测试对于图层并调试其中包含test.cpp的{​​{1}}并调用新图层？我应该写一些类似于main中的图层测试文件吗？

Answer 1

您绝对应该为您的图层编写测试，测试forward和渐变计算（使用自动数值梯度测试工具）。
完成此测试后，可以直接确定您的图层是否正常工作，如果没有，则在不运行整个框架的情况下更容易调试。