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.
if you change
to
The multi gpu setups can run. Also, then
the generator is the only place where the batch size is specified.