Sure sir,
Inputs shape : torch.Size([24, 20000, 100])
Output shape : torch.Size([480000, 6])
Labels shape : torch.Size([480000])
Agreement score shape: torch.Size([480000])
Can it be a cuda error?
I perform model training in cuda and I compute the loss and accuracy on cpu.