Pytorch 1.6.0 - RuntimeError: CUDA error: device-side assert triggered where (predicted==labels)

Question

When I run this function:

def evaluate(model, dataloader, calc_loss=False):
\'\'\'
Helper function to get classification accuracy and (optional) loss for a model