SeeSR icon indicating copy to clipboard operation
SeeSR copied to clipboard

Published 20 hours ago • verified publisher icon cswry

About the GPU memory usage

Open prokaryote-hgy opened this issue 7 months ago • 4 comments

Hi, wonderful work!

I wonder what's the peak memory usage when training the model with HR size of 512 and batchsize of 1 for one GPU. It seams that a 24GB GPU not enough to run the training? Or I did something wrong?

prokaryote-hgy avatar Jul 14 '24 13:07 prokaryote-hgy