OOM(out of memory) on 2080Ti (with 11G memory) when run test.py
Jiangyu1181 opened this issue · 1 comments
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="$proj_root/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
half 'n_rays_per_step' of '.ini' file of 'config'