rishizek/tensorflow-deeplab-v3

Multiple GPUs for training

DarylWM opened this issue · 2 comments

DarylWM commented

Thanks for sharing this code @rishizek.

Has anyone managed to train this on more than one GPU?

John1231983 commented

Hi, I tried it but it was not successful. The reason is that training with batch norm in multiple GPU. It is not supported in tensorflow yet

DarylWM commented

Ok, thanks.