RuntimeError: CUDA error: an illegal memory access was encountered

wgcban / ddpm-cd

Remote Sensing Change Detection using Denoising Diffusion Probabilistic Models

https://www.wgcban.com/research#h.ar24vwqlm021

MIT License

247 stars 29 forks source link

RuntimeError: CUDA error: an illegal memory access was encountered #1

Closed Crawlling closed 1 year ago

Crawlling commented 1 year ago

when i trainning ddpm that code run to line "diffusion.optimize_parameters()",then i got this issue,i saw "# need to average in multi-gpu" in the function "optimize_parameter",but i don't know how to do.please help.