liangzid:

this a problem in my codes:


thanks,i have found the thing going wrong, the last torch.Size is not same as before.
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 2048])
***************************************************************
torch.Size([2, 1024])