It seems as if one of the passed tensors was passed as an int instead.
Could you check that?
Also, based on your print statement I’m not sure, if label is a list or a tensor.
Anyway, it should be a tensor with the same shape as your model’s output.