Focal-Loss icon indicating copy to clipboard operation
Focal-Loss copied to clipboard

break when save model

Open jjkke88 opened this issue 7 years ago • 2 comments

when I save model, the script crash , the usage of focalloss in my code is below

net.loss = L.FocalLoss(net.deconv, net.label, loss_weight=1, propagate_down=[1, 0], loss_param=dict(normalize=True, ignore_label=-1), focal_loss_param=dict(alpha=0.25, gamma=2))

jjkke88 avatar Aug 23 '17 02:08 jjkke88

hi @jjkke88,

which framework do you use?

zimenglan-sysu-512 avatar Aug 23 '17 10:08 zimenglan-sysu-512

caffe

jjkke88 avatar Aug 25 '17 14:08 jjkke88