Output Batch has a size:
torch.Size([12, 12, 3])
Target Batch has a size
torch.Size([12, 1])
I change a line in def forward: to
class_label = self.out(rnn_out.contiguous().view(-1, self.hidden_size))
to meet input expectation of linear
now,
Output Batch has a size:
torch.Size([144, 3])
I want it to be [12, 3]