diff --git a/ppocr/losses/combined_loss.py b/ppocr/losses/combined_loss.py index 639265ede9166d2558bf3d9290ffe7eefa55ac27..eb1ad9bf92b53ada4ba32b4ab238e3f9f3f6ae22 100644 --- a/ppocr/losses/combined_loss.py +++ b/ppocr/losses/combined_loss.py @@ -53,8 +53,9 @@ class CombinedLoss(nn.Layer): for key in loss: if key == "loss": loss_all += loss[key] * weight - # else: - # loss[f"{key}_{idx}"] = loss[key] + else: + loss["{}_{}".format(key, idx)] = loss[key] + # loss[f"{key}_{idx}"] = loss[key] loss_dict.update(loss) loss_dict["loss"] = loss_all return loss_dict