CUDA out of memory - on 12 GB GPU

showlab / DatasetDM

[NeurIPS2023] DatasetDM:Synthesizing Data with Perception Annotations Using Diffusion Models

https://weijiawu.github.io/DatasetDM_page/

290 stars 12 forks source link

Open jonistuetz opened 4 months ago

jonistuetz commented 4 months ago

Are there any ways to optimize the process so that the OOM error can be avoided?