CUDA out of memory
SunshineBread opened this issue · comments
SunshineBread commented
Hi, I’m not very familiar with the code structure of this project. When I run the scripts in the readme, I encounter a ‘cuda out of memory’ error. Could you please advise me on how to modify the code so that the program can run?
Yiwu Zhong commented
@SunshineBread You can try reducing the batch size. For example, the parameter in any config file.