Segmentation using "softmaxloss/vl_nnloss"

[HosnaCSE]

Hello,

I am trying to segment medical image using "softmaxloss" which refer to "vl_nnloss". I can segment my image using this example. But when I use "softmaxloss" layer, removing the last layer (forward and backward pass) as follows:

%remove this
net = addCustomLossLayer(net, @l2LossForward, @l2LossBackward) ;

%add this
net.layers{end+1} = struct(...
'name', 'loss', ...
'type', 'softmaxloss') ;

But my objective function generate error '0'. I couldn't figure out what else I should do to make this work with "softmaxloss/vl_nnloss" instead of regression. I will appreciate any kind of help.

Sincerely
Hosna