search

Oliiveralien / MDMS

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

关于显存 #5

Closed Mayingj closed 7 months ago

Mayingj commented 7 months ago

请问你们用的显存是多大?我在文章中Training details没有找到对显卡的说明，我的显存是24G,但是在test的时候，依旧会显存溢出。

skkkyup commented 7 months ago

多尺度设计为64×64，96×96，128×128的块，步长为8，对于LOL数据集运行起来需要40G左右，实际使用的是A40 48G显卡。