CUDA_ERROR_OUT_OF_MEMORY
Tinky7 opened this issue · comments
Tinky7 commented
How to use multiple GPUs for training when there is insufficient graphics memory during neus2 training?
Yiming Wang 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.