caffe layer层cpp、cu调试经验和相互关系

对于layer层的cpp文件,你可以用LOG和printf、cout进行调试,cu文件不能使用LOG,可以使用cout,printf。

对于softmaxloss的layer层,既有cpp文件又有cu文件,其中cu文件只实现了前向和反向传播,cpp文件既实现了前向、反向,也实现了layer setup,layer reshape,还有一些定义的函数。我用上述方法进行调试发现,在gpu条件下,cpp的前向和反向不使用,但其他函数依然使用,gpu下的前向反向是使用cu中的代码。

posted @ 2018-02-27 15:51  有梦就要去实现他  阅读(596)  评论(0编辑  收藏  举报