vincentfung13 / MINE

Code and models for our ICCV 2021 paper "MINE: Towards Continuous Depth MPI with NeRF for Novel View Synthesis"
MIT License
406 stars 43 forks source link

minimum hardward requirements #1

Closed emjay73 closed 3 years ago

emjay73 commented 3 years ago

Thank you for your nice work!

What if I want to run your code, do I need 48 V100 GPUs as you mentioned in the paper?

What are the minimum requirements to run this code?

Thanks in advance.

vincentfung13 commented 3 years ago

Hi, thank you for your interest in our work. The 48 V100 GPUs setting was for training the large RealEstate10K dataset. For running the LLFF training provided in the current code and the demo, a couple of GPUs at most should do.

lijx10 commented 3 years ago

It takes only 1 GPU, e.g., 2080ti, to run network inference and the demo we provided. You may need to tune the batch size and/or input image size if there is out-of-memory problem at inference.

emjay73 commented 3 years ago

Thanks, Just for information, can you briefly note the training time required for each dataset? (with number of GPUs)

vincentfung13 commented 3 years ago

Note that these numbers depend on the specific hyperparams you choose for your training (such as input size and the number of planes), and you might need to tune it a little bit yourself by looking at the validation results.

emjay73 commented 3 years ago

OK, That seems doable now. 🤣 I'll check the hyperparameters that you mentioned later. Thank you!