guillaumegenthial/im2latex

gpu train?

pageedward opened this issue · 1 comments

i set batch_size=256,but mem use only 309MB,

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?