gpu train?
pageedward opened this issue · 1 comments
pageedward commented
i set batch_size=256,but mem use only 309MB,
Root970103 commented
Hi, I also set the multi-gpu, but it semms that only gpu0 is used, other three gpus only use 393M. Have you found out the cause?