diff --git a/ppocr/modeling/heads/rec_att_head.py b/ppocr/modeling/heads/rec_att_head.py index 7926f0741f51135760cc90b6419d13bafa0cbaee..ab8b119fe08ac79a6b98a449bb117da018df2ff3 100644 --- a/ppocr/modeling/heads/rec_att_head.py +++ b/ppocr/modeling/heads/rec_att_head.py @@ -45,7 +45,6 @@ class AttentionHead(nn.Layer): output_hiddens = [] if targets is not None: - print("target is not None") for i in range(num_steps): char_onehots = self._char_to_onehot( targets[:, i], onehot_dim=self.num_classes) @@ -55,7 +54,6 @@ class AttentionHead(nn.Layer): output = paddle.concat(output_hiddens, axis=1) probs = self.generator(output) else: - print("target is None") targets = paddle.zeros(shape=[batch_size], dtype="int32") probs = None char_onehots = None