x, endpoints = self._mobilenet_base(inputs=inputs)
endpoints[max(endpoints.keys()) + 1] = x
self._output_specs = {l: endpoints[l].get_shape() for l in endpoints}
super(MobileNet, self).__init__(
After Change
x, endpoints, next_endpoint_level = self._mobilenet_base(inputs=inputs)
endpoints[str(next_endpoint_level)] = x
self._output_specs = {l: endpoints[l].get_shape() for l in endpoints}
super(MobileNet, self).__init__(