如何将自定义丢失函数传递给net.backward()

时间:2016-04-18 10:49:07

标签: machine-learning neural-network caffe

Caffe注意到

  

向后传球计算给定学习损失的梯度[...]

但是你可以打电话

net.backward() 

仍然得到差异甚至没有传递任何预期的输出。我想根据输入和我自己的损失函数计算梯度。我如何在Caffe中做到这一点?

0 个答案:

没有答案