training requirement of GPU memory

jchibane / srf

Stereo Radiance Fields (SRF) - Codebase

123 stars 18 forks source link

training requirement of GPU memory #11

Closed hengfei-wang closed 2 years ago

hengfei-wang commented 2 years ago

Hi,

Thank you for releasing the code.

I am trying to train this model on my dataset. I just revised the dataset-setup code. The resolution of my image is 2000x2000. and I set ref_imgs number to 5. The GPU is RTX 2080 with 12GB memory. I try to train this model several times, but I got the following err message:

So I want to ask the requirement of GPU memory for training this model.

Best, Hengfei

hengfei-wang commented 2 years ago

I see that the gpu memory of the paper setting is 48GB.

closed