Cuda Error in scripts
cabisarri opened this issue · comments
Hi, I am just trying your example notebook "examples/notebook_examples/sst2.ipynb". It states that distiller_basic and distiller_utils will encounter CUDA error: device-side assert triggered. I guess it is the way you designed to do batch.to(device), however, the device is not defined in either GeneralDistiller, they wont receive it.