search

19reborn / NeuS2

[ICCV 2023] Official code for NeuS2
Other
641 stars 42 forks source link

CUDA_ERROR_OUT_OF_MEMORY #51

Closed Tinky7 closed 9 months ago

Tinky7 commented 11 months ago

How to use multiple GPUs for training when there is insufficient graphics memory during neus2 training?

19reborn commented 10 months ago

Hi @Tinky7 , thanks for using our code. The current code deosn't support multiple GPUs for training. You can decrease the training batch size by setting batch_size value in the config file.