cuda out of memory

[CCOSerika]

Hi, author. Thanx for your great work!
I cannot wait to see how it will work. I train on my RTX3090 but it gives "cuda out of memory".
Could u kindly provide the trained ckpts with us to check? Or how can I set the parameters to run it successfully.
Looking forward to your reply!

[CCOSerika]

By the way, what graphics card did u use for training? Will RTX3090 be okey?

[MukundVarmaT]

Thank you for your interest in our work. We have released the pre-trained models here. We run all our experiments on RTX A6000 cards (with 48G memory). You could try reducing the N_rand parameter to train the model on smaller cards.

[CCOSerika]

I got it. Thank you so much for your reply. I'll try it.