GPU_Variable_shape_debug

Author: wabywang(王本友)

models/Transformer.py

@@ -425,7 +425,7 @@ def forward(self, inp):
          src_seq,src_pos = inp
 #         enc_output, *_ = self.encoder(src_seq, src_pos)   #64x200x512
          enc_output = self.encoder(src_seq, src_pos)   #64x200x512
-         return  self.hidden2label(enc_output.view((enc_output.shape[0],-1)))
+         return  self.hidden2label(enc_output.view((self.batch_size,-1)))