How to penalize a category error more than others categories during training (SoftMaxWithLoss layer)

35 views
Skip to first unread message

Hermann Hesse

unread,
Jul 6, 2016, 7:20:35 AM7/6/16
to Caffe Users
Hi all,

is there any way to penalize more a category error during training the network? For example, accumulating two times more than standard for a category error when computing the loss.

I'm using SoftMaxWith(cross-entropy)Loss layer, but I can use another cost function if necessary. I guess that i should edit the softmax_loss_layer.cu file, but my CUDA level is quite low.

Thanks so much.

Hermann Hesse

unread,
Jul 18, 2016, 8:56:13 AM7/18/16
to Caffe Users
Anyone please?
Reply all
Reply to author
Forward
0 new messages