Tensorflow For Image Segmentation: Changing Minibatch Size Stops Learning

I have created a net for image segmentation, in particular brain tumors. The jupyter notebook with the code is here. When I train the CNN with minibatch size of 1, I get a fairly g

Solution 1:

I found the problem. I checked my graph with tensorboard and I notice that in the CONV1/S1 I was not connecting the output of the ReLu to the next layer (CONV1/S2), instead I was connecting the output of the conv2d directly.

I changed that line in the code and everything is working as expected.