Try to use more than one GPU get errors

sachinraja13 / TabStructNet

Other

129 stars 32 forks source link

Try to use more than one GPU get errors #10

Open wangwen-whu opened 3 years ago

wangwen-whu commented 3 years ago

Could you help me deal this problem? I change the GPU_COUNT=2 in tabnet.py, however i got this error seems like batch_size couldn't match the code. But if i change Image_PER_GPU=2 (or4) that is ok,I can train this code.
Could you help me how can I use more than one GPU?

anishdulal commented 3 years ago

Hi @wangwen-whu, I am facing similar issue as you do, were you able to find the solution?