Skip to content
Snippets Groups Projects
Commit d4c505d7 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!199 Fix dim of ReverseV2 in CRNN

Merge pull request !199 from chenhaozhe/fix-crnn-reverse
parents 48b6e8f6 f0541d31
Branches
Tags
No related merge requests found
......@@ -231,8 +231,8 @@ class CRNNV1(nn.Cell):
self.transpose = P.Transpose()
self.squeeze = P.Squeeze(axis=0)
self.vgg = VGG()
self.reverse_seq1 = P.ReverseV2(axis=[1])
self.reverse_seq2 = P.ReverseV2(axis=[1])
self.reverse_seq1 = P.ReverseV2(axis=[0])
self.reverse_seq2 = P.ReverseV2(axis=[0])
self.seq_length = Tensor(np.ones((self.batch_size), np.int32) * config.num_step, mstype.int32)
self.concat1 = P.Concat(axis=2)
self.rnn_dropout = nn.Dropout(0.9)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment