Try to use more than one GPU get errors

[wangwen-whu]

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]

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