A library for experimenting with, training and evaluating neural networks, with a focus on adversarial robustness.
903
stars
181
forks
source link
Training on Custom Dataset triggers error : CUDA error: device-side assert triggered #97
Open
elcronos opened 3 years ago
I'm using your Python package to adversarially train a ResNet50 network for CelebA dataset. My code looks something like this:
My definition for CelebA in robustness/datasets.py is:
Once I run my code it starts training as expected but then after a while I get this error:
In addition, when I try to change the batch_size I get the error:
Any insight on why is this happening and how to solve it?