Thanks, I figure out the problem. For some reason (Beyond me), some examples had a class that was n_classes + 1, so it was giving this error when your batch contained those examples. I redownloaded the dataset and it’s working now.