19reborn / NeuS2

[ICCV 2023] Official code for NeuS2

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CUDA_ERROR_OUT_OF_MEMORY

Tinky7 opened this issue · comments

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

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.