ValueError: Tensor's shape (1, 1, 128, 18) is not compatible with supplied shape (1, 1, 128, 36) in Object Detection API

Question

I am trying to build an object detection model, using the object detection api, but while trying to predict the classes I am getting this error upon running the detect_fn(in