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

(1)进行了函数内容缩进(2)加上了alpha前面忘加的self.(3)loss的求和方式增加了一种batchsize方式

Open zxcvbnmkj opened this issue 7 months ago • 0 comments

您的代码真的非常好,简洁易懂,教程详细,非常感谢您的无私分享,但是我运行时注意到代码尚且存在两点小问题【(1)与(2)】所以特此发起了Pull 请求 (1)def forward(self, preds, labels):函数中的内容您忘了缩进了 (2)第45行: alpha = self.alpha.to(preds.device)忘记在alpha前面加上self了,使用cuda时会报错--alpha和label不在一个设备上。 (3)原先的loss返回方式只有sum和mean两种,我新增加了一种batch_size方式。

zxcvbnmkj avatar Jun 30 '24 12:06 zxcvbnmkj