From 5613e21d860da9e50dbacae6686b2df49152d692 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Tue, 28 Sep 2021 18:27:44 +0800 Subject: [PATCH] fix map name (#4191) --- ppocr/losses/distillation_loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/losses/distillation_loss.py b/ppocr/losses/distillation_loss.py index 73d3ae2a..06aa7fa8 100644 --- a/ppocr/losses/distillation_loss.py +++ b/ppocr/losses/distillation_loss.py @@ -112,7 +112,7 @@ class DistillationDMLLoss(DMLLoss): if isinstance(loss, dict): for key in loss: loss_dict["{}_{}_{}_{}_{}".format(key, pair[ - 0], pair[1], map_name, idx)] = loss[key] + 0], pair[1], self.maps_name, idx)] = loss[key] else: loss_dict["{}_{}_{}".format(self.name, self.maps_name[ _c], idx)] = loss -- GitLab