NVIDIA / caffe

Caffe: a fast open framework for deep learning.
http://caffe.berkeleyvision.org/
Other
672 stars 263 forks source link

how to add centerLoss_layer #547

Closed happy-Bobo closed 5 years ago

happy-Bobo commented 5 years ago

Hello I want to train my model with nvcaffe , but my net contain centerLoss layer, so I failed. I am try to add the 'center_Loss_layer.cpp/hpp/cu' to the src/include, but build the framework error, would you help me solve this question, thank you very much! center_loss_layer.cpp.txt center_loss_layer.cu.txt center_loss_layer.hpp.txt