csguoh / MambaIR

[ECCV2024] An official pytorch implement of the paper "MambaIR: A simple baseline for image restoration with state-space model".

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The requirements for GPU memory?

CubicsYang opened this issue · comments

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🤣

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