"Dimension 0 in both shapes must be equal,but are 4 and 3"

[yan4821567]

I run the code with patch '128x128x128', and cuda out of memory. So I change patch size to '20x128x128'，and it says that:

ValueError: Dimension 0 in both shapes must be equal, but are 4 and 3. Shapes are [4,16,16] and [3,16,16]. for 'tower0/unet3d/concat' (op: 'ConcatV2') with input shapes: [1,128,4,16,16], [1,128,3,16,16], [] and with computed input tensors: input[2] = <1>.

But when I change patch size to '64x64x64', it works. But I am not sure if this is the best setting.