I have a question about GPU capabilities.
Closed this issue · 2 comments
KATSU-HIGH5 commented
I'm trying to train VRCNet, but it runs out of RAM when the batch size is more than 16.
The GPU I am using is a GTX 1080 with 8GB.
What kind of GPU are you using and what batch size are you training?
paul007pl commented
For most experiments, we use a V100 GPU (32GB) with batch size 32.
If you are using other GPUs with smaller memory, you can try a smaller batch size.
According to my experience, it won't affect a lot.
KATSU-HIGH5 commented
Thank you very much!!