Closed ChoiIseungil closed 3 years ago
Whenever I try Res50, I got cuda out of memory error. Is this problem happening because the model is too large to handle with a single GPU? To train such a large model, do I have to parallelize code to use multiple sessions?
Whenever I try Res50, I got cuda out of memory error. Is this problem happening because the model is too large to handle with a single GPU? To train such a large model, do I have to parallelize code to use multiple sessions?