THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=383 error=11 : invalid argument

Raschka-research-group / coral-cnn

Rank Consistent Ordinal Regression for Neural Networks with Application to Age Estimation

MIT License

341 stars 62 forks source link

Closed NarasimmanSaravana1994 closed 4 years ago

NarasimmanSaravana1994 commented 4 years ago

While training... I'm using CUDA 10 and pytorch version 1.1.0. Is there ary issue in my side.

NarasimmanSaravana1994 commented 4 years ago

While upgrade my pytorch with CUDA 10 stable version my issue was fixed.