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