Error while training with SSD Mobilenet

[bored-wolf]

When I run the code for training with the SSD Mobilenet model, I get the following error:

ValueError: Shape must be rank 1 but is rank 4 for 'conv1_bn/cond/FusedBatchNorm' (op: 'FusedBatchNorm') with input shapes: [?,150,150,32], [1,150,1,1], [1,150,1,1], [1,150,1,1], [1,150,1,1]

for an input shape of (300,300,3)

Is there any change that would need to be made for rectifying this?