diff --git a/ppocr/losses/combined_loss.py b/ppocr/losses/combined_loss.py index eb1ad9bf92b53ada4ba32b4ab238e3f9f3f6ae22..f37c4db128e5b0cb3ef9a9fe249402ba37d5ceeb 100644 --- a/ppocr/losses/combined_loss.py +++ b/ppocr/losses/combined_loss.py @@ -17,7 +17,7 @@ import paddle.nn as nn from .distillation_loss import DistillationCTCLoss from .distillation_loss import DistillationDMLLoss -from .distillation_loss import DistillationDistanceLoss +from .distillation_loss import DistillationDistanceLoss, DistillationDBLoss, DistillationDilaDBLoss class CombinedLoss(nn.Layer):