search

Oliiveralien / MDMS

Multi-Domain Multi-Scale Diffusion Model for Low-Light Image Enhancement (AAAI'24)
23 stars 0 forks source link

how much gpu memory need?需要多大的显存 #7

Closed lightning0016 closed 6 months ago

lightning0016 commented 6 months ago

i use a30 GPU 24G memory,no other app run,always outmem.

skkkyup commented 6 months ago

Since the stride for multi-scale sampling is set to 8, a large number of blocks are generated. We are using a 48G A40 GPU. If you want to run it on a 24G GPU, you can increase the stride appropriately to reduce the GPU requirements.