search

csguoh / MambaIR

[ECCV2024] An official pytorch implement of the paper "MambaIR: A simple baseline for image restoration with state-space model".
Apache License 2.0
361 stars 33 forks source link

The requirements for GPU memory? #38

Open CubicsYang opened 3 weeks ago

CubicsYang commented 3 weeks ago

I would like to know what the minimum GPU memory is. I plan to complete a single channel image super-resolution task but don't have 8 GPUs🤣

csguoh commented 3 weeks ago

Two 24G 3090 can work well with MambaIR-light model.