mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

fix: there may not be weight ratio in multilabel dataset label (#3226)

Browse Source
This commit is contained in:
Tingquan Gao 2024-08-23 11:35:56 +08:00 committed by GitHub
parent 666ca9bbd1
commit bc9401f861
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/loss/multilabelloss.py
View File

@ -39,7 +39,7 @@ class MultiLabelLoss(nn.Layer):
def _binary_crossentropy(self, input, target, class_num):
if self.weight_ratio:
target, label_ratio = target[:, 0, :], target[:, 1, :]
else:
elif target.ndim == 3:
target = target[:, 0, :]
if self.epsilon is not None:
target = self._labelsmoothing(target, class_num)