google / nerfactor

Neural Factorization of Shape and Reflectance Under an Unknown Illumination
https://xiuming.info/projects/nerfactor/
Apache License 2.0
437 stars 56 forks source link

OOM(out of memory) on 2080Ti (with 11G memory) when run test.py #15

Closed Jiangyu1181 closed 2 years ago

Jiangyu1181 commented 2 years ago

I really appreciate your work and open source, but when I run the following code, I get an ERR: OOM

III. Simultaneous Relighting and View Synthesis (testing)

‘’‘ scene='hotdog_2163' gpus='4,5,6,7' model='nerfactor' overwrite='True'

proj_root='/mnt/data1/jy/NeRFactor' repo_dir="$proj_root/nerfactor" # /mnt/data1/jy/NeRFactor/nerfactor/ outroot="$projroot/output/train/${scene}$model" viewer_prefix='http://vision38.csail.mit.edu' # or just use '' ckpt="$outroot/lr5e-3/checkpoints/ckpt-10" if [[ "$scene" == pinecone || "$scene" == vasedeck || "$scene" == scan* ]]; then

Real scenes: NeRF & DTU

color_correct_albedo='false'

else color_correct_albedo='true' fi REPO_DIR="$proj_root" "$proj_root/nerfactor/test_run.sh" "$gpus" --ckpt="$ckpt" --color_correct_albedo="$color_correct_albedo" ‘’’

My GPU is 2080Ti (with 11G memory) , when I want to lower batch size, I find 'no_batch = True' in 'lr5e-3.ini'. So how can I successfully run 'test.py' on 2080Ti.

Very much looking forward to your help

Jiangyu1181 commented 2 years ago

half 'n_rays_per_step' of '.ini' file of 'config'