Problem while retraining CNN models with pre-trained model

[prashanthvs]

I am getting this error when retraining a custom database. I followed the exact requirements.txt and setup the necessary environment, but unable to understand this error.

ValueError: Dimension 0 in both shapes must be equal, but are 7 and 16 for 'Assign' (op: 'Assign') with input shapes: [7,7,1,16], [16,3,7,7]