Brian_Mullen:

didn’t match because some of the arguments have invalid types: (int, int, torch.FloatTensor, torch.cuda.FloatTensor)


Your model is not transferred to CUDA maybe. Try adding the line: model.cuda()