noveens:

for x, y in train_reader.iter():
print x.shape # input shape is "bsz x seq_len"
hidden_enc, hidden_dec = model(x, hidden_enc, hidden_dec)
print output.shape # output shape is “bsz/num_gpu x seq_len”


How did you get output? It’s not clear how it’s generated from this code.