Thanks for the update. Would it be possible to provide the model definition, so that we could try to reproduce this issue?
I don’t think it’s related to the dataset, but might be an internal cudnn error.