MVSNet icon indicating copy to clipboard operation
MVSNet copied to clipboard

OOM error when trying example R-MVSNet first example

Open Codeguyross opened this issue 5 years ago • 1 comments

Hello,

I am trying to run the first example for R-MVSNet and I am having a memory issue. The command I am trying to run is: python train.py --regularization 'GRU' --train_blendedmvs --max_w 768 --max_h 576 --max_d 128 --batch_size 1 --online_augmentation

The error I am receiving is:

I am using the example data from the MVSNet link: https://drive.google.com/open?id=1ilxls-VJNvJnB7IaFj7P0ehMPr7ikRCb

I have an 8gb video card with around 7.3gb available for useand would like settings that would allow me to run using what I have available.

Thanks for the help!

Codeguyross avatar May 11 '20 17:05 Codeguyross

running the following on 8gb gpu has worked for me with material provided by author , example using '3DCNNs' for regularization would generate Out of memory error when using < 12gb card

python test.py --dense_folder TEST_DATA_FOLDER --regularization 'GRU' --max_w 1600 --max_h 1200 --max_d 256 --interval_scale 0.8 --pretrained_model_ckpt_path ~/MVSNet/mvsnet/data/GRU/model.ckpt --ckpt_step 150000

BingEdison avatar May 13 '20 03:05 BingEdison