paul007pl/VRCNet

I have a question about GPU capabilities.

Closed this issue · 2 comments

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?

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.

Thank you very much!!