About the training resource

Question

About the training resource

cskkxjk opened this issue 8 months ago · comments

Hi~ Thanks for your great work!
I wonder what kind of GPUs are required to train with the provided config config_files/ms_blender/TriMipRF.gin ?
I failed on 1080 due to some memory issues.

Hu Wenbo · Answer 1 · Fri Oct 13 2023 20:13:25 GMT+0800 (China Standard Time)

We tested it on A100-80g, but it only consumes about 9GB gpu memeory. If it still exceeds your gpu memory, please consider tune down Trainer.target_sample_batch_size

XuJunKaiZJU · Answer 2 · Fri Oct 13 2023 21:42:23 GMT+0800 (China Standard Time)

We tested it on A100-80g, but it only consumes about 9GB gpu memeory. If it still exceeds your gpu memory, please consider tune down Trainer.target_sample_batch_size

Thank you for the reply!

Flyingdog-Huang · Answer 3 · Mon Nov 13 2023 11:43:14 GMT+0800 (China Standard Time)

We tested it on A100-80g, but it only consumes about 9GB gpu memeory. If it still exceeds your gpu memory, please consider tune down Trainer.target_sample_batch_size

hi，I try to change the value of Trainer.target_sample_batch_size, change to large or small from 262144, but the GPU memory did not change, which parameter can control GPU-men?
thanks a lot!