19reborn/NeuS2

CUDA_ERROR_OUT_OF_MEMORY

Closed this issue · 1 comments

Tinky7 commented

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

19reborn commented

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.