search

rishizek / tensorflow-deeplab-v3

DeepLabv3 built in TensorFlow
MIT License
286 stars 102 forks source link

Multiple GPUs for training #2

Closed DarylWM closed 6 years ago

DarylWM commented 6 years ago

Thanks for sharing this code @rishizek.

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

John1231983 commented 6 years ago

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 6 years ago

Ok, thanks.