Update rec_nrtr_optim_head.py

ppocr/modeling/heads/rec_nrtr_optim_head.py

@@ -216,7 +216,7 @@ class TransformerOptim(nn.Layer):
             new_shape = (n_curr_active_inst * n_bm, *d_hs)
 
             beamed_tensor = beamed_tensor.reshape(
-                [n_prev_active_inst, -1])  #contiguous()
+                [n_prev_active_inst, -1])
             beamed_tensor = beamed_tensor.index_select(
                 paddle.to_tensor(curr_active_inst_idx), axis=0)
             beamed_tensor = beamed_tensor.reshape([*new_shape])
@@ -337,7 +337,7 @@ class TransformerOptim(nn.Layer):
             n_inst, len_s, d_h = src_enc.shape
             src_enc = paddle.concat([src_enc for i in range(n_bm)], axis=1)
             src_enc = src_enc.reshape([n_inst * n_bm, len_s, d_h]).transpose(
-                [1, 0, 2])  #repeat(1, n_bm, 1)
+                [1, 0, 2])
             #-- Prepare beams
             inst_dec_beams = [Beam(n_bm) for _ in range(n_inst)]