fix map name (PaddlePaddle#4191)

bupt906 · Sep 28, 2021 · 5613e21 · 5613e21
1 parent 773a8c4
commit 5613e21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppocr/losses/distillation_loss.py b/ppocr/losses/distillation_loss.py
@@ -112,7 +112,7 @@ def forward(self, predicts, batch):
                     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