Skip to content

GitLab

Commit 9811a09e authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix combined loss

parent 48898ac3
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
ppocr/losses/combined_loss.py
View file @ 9811a09e
...@@ -53,8 +53,9 @@ class CombinedLoss(nn.Layer): ...@@ -53,8 +53,9 @@ class CombinedLoss(nn.Layer):
for key in loss: for key in loss:
if key == "loss": if key == "loss":
loss_all += loss[key] * weight loss_all += loss[key] * weight
# else: else:
# loss[f"{key}_{idx}"] = loss[key] loss["{}_{}".format(key, idx)] = loss[key]
# loss[f"{key}_{idx}"] = loss[key]
loss_dict.update(loss) loss_dict.update(loss)
loss_dict["loss"] = loss_all loss_dict["loss"] = loss_all
return loss_dict return loss_dict
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment