search

advaitsave / Multiclass-Semantic-Segmentation-CamVid

Tensorflow 2 implementation of complete pipeline for multiclass image semantic segmentation using UNet, SegNet and FCN32 architectures on Cambridge-driving Labeled Video Database (CamVid) dataset.
82 stars 41 forks source link

batch_shape causes multi gpu to fail #4

Open johngrabner opened 4 years ago

johngrabner commented 4 years ago

if you change 

batch_shape=(256,256,3)
inputs = Input(batch_shape=(5, 256, 256, 3))

to

shape=(256,256,3)
inputs = Input(shape)

The multi gpu setups can run. Also, then the generator is the only place where the batch size is specified.