fxcdl opened this issue a year ago · 0 comments
Hi, thank u for your code, I try to train grl_base model on dn task(patch_size=256, stripe_size1=64, stripe_size=128, batch_size=1)，but gpu(3090(24GB)✖4) is out of memory, how much gpu memory i need to train this model.