Skip to content

GitLab

Unverified Commit ddbba2b1 authored by zhoujun's avatar zhoujun Committed by GitHub
Browse files

Merge pull request #2228 from WenmuZhou/fix_attn_export 

fix attn export
parents f687e092 592c4187
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/modeling/heads/rec_att_head.py
View file @ ddbba2b1
...@@ -38,7 +38,7 @@ class AttentionHead(nn.Layer): ...@@ -38,7 +38,7 @@ class AttentionHead(nn.Layer):
return input_ont_hot return input_ont_hot
def forward(self, inputs, targets=None, batch_max_length=25): def forward(self, inputs, targets=None, batch_max_length=25):
batch_size = inputs.shape[0] batch_size = paddle.shape(inputs)[0]
num_steps = batch_max_length num_steps = batch_max_length
hidden = paddle.zeros((batch_size, self.hidden_size)) hidden = paddle.zeros((batch_size, self.hidden_size))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment