RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)` free(): invalid next size (fast)

mahmoudnafifi / HistoGAN

Reference code for the paper HistoGAN: Controlling Colors of GAN-Generated and Real Images via Color Histograms (CVPR 2021).

MIT License

267 stars 27 forks source link

RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)` free(): invalid next size (fast) #21

Open wkdgnsgo opened 2 years ago

wkdgnsgo commented 2 years ago

Hi I tried to train the network with FFHQ data set and I faced this error. I searched the error and found that some of layers' dimensions got mismatched. Have you guys met this problem before? Thanks

mahmoudnafifi commented 2 years ago

I didn't see this error before. Sorry I cannot help on this. Let's keep the issue open hopefully anyone else can suggest a fix.