zimenglan-sysu-512 / Focal-Loss

loss layer of implementation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to replace FocalLoss in testing phase?

SingL3 opened this issue · comments

In training phase, we use SoftmaxWithLoss Layer and replace it with Softmax in testing. If I use FocalLoss in training, is it still replaced by Softmax?