Skip to content

Commit de45732

Browse files
reset latex ocr (#14046)
1 parent de82b7b commit de45732

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ppocr/modeling/heads/rec_latexocr_head.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -948,15 +948,15 @@ def generate_export(
948948
b, t = start_tokens.shape
949949

950950
self.net.eval()
951-
out_tmp = start_tokens
951+
out = start_tokens
952952
mask = kwargs.pop("mask", None)
953953

954954
if mask is None:
955-
mask = paddle.full_like(out_tmp, True, dtype=paddle.bool)
955+
mask = paddle.full_like(out, True, dtype=paddle.bool)
956956

957957
i_idx = paddle.full([], 0)
958958
while i_idx < paddle.to_tensor(seq_len):
959-
x = out_tmp[:, -self.max_seq_len :]
959+
x = out[:, -self.max_seq_len :]
960960
paddle.jit.api.set_dynamic_shape(x, [-1, -1])
961961
mask = mask[:, -self.max_seq_len :]
962962
paddle.jit.api.set_dynamic_shape(mask, [-1, -1])
@@ -969,7 +969,7 @@ def generate_export(
969969
probs = F.softmax(filtered_logits / temperature, axis=-1)
970970

971971
sample = paddle.multinomial(probs, 1)
972-
out = paddle.concat((out_tmp, sample), axis=-1)
972+
out = paddle.concat((out, sample), axis=-1)
973973

974974
pad_mask = paddle.full(shape=[mask.shape[0], 1], fill_value=1, dtype="bool")
975975
mask = paddle.concat((mask, pad_mask), axis=1)

0 commit comments

Comments
 (0)