From 30d4d5eedbe7ecbe84cd7ff4585d4dff74838972 Mon Sep 17 00:00:00 2001 From: panfengfeng Date: Thu, 29 Jul 2021 09:04:02 +0800 Subject: [PATCH] Loss to LossBase --- model_zoo/official/cv/crnn_seq2seq_ocr/src/attention_ocr.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model_zoo/official/cv/crnn_seq2seq_ocr/src/attention_ocr.py b/model_zoo/official/cv/crnn_seq2seq_ocr/src/attention_ocr.py index c789cd3fce1..172867b4b1b 100755 --- a/model_zoo/official/cv/crnn_seq2seq_ocr/src/attention_ocr.py +++ b/model_zoo/official/cv/crnn_seq2seq_ocr/src/attention_ocr.py @@ -24,7 +24,7 @@ import mindspore.nn as nn import mindspore.ops.operations as P import mindspore.common.dtype as mstype from mindspore import context, Tensor -from mindspore.nn.loss.loss import Loss +from mindspore.nn.loss.loss import LossBase from mindspore.ops import composite as C from mindspore.ops import functional as F from mindspore.context import ParallelMode @@ -34,7 +34,7 @@ from mindspore.parallel._auto_parallel_context import auto_parallel_context from src.seq2seq import Encoder, Decoder -class NLLLoss(Loss): +class NLLLoss(LossBase): def __init__(self, reduction='mean'): super(NLLLoss, self).__init__(reduction) self.one_hot = P.OneHot()