Hi, I tired what you suggested. I got this error.
ERR: Target and input must have the same number of elements. target nelement (100) != input nelement (200)
I also printed the shapes I am getting. My batch size is 100, and number of layers 3.
last_h torch.Size([3, 2, 100, 64])
final_h torch.Size([2, 100, 64])
fc_out torch.Size([2, 100, 1])