I am getting an error converting a tf estimator saved model to model.json and weight shards format. I am not able to understand where the error is. Any pointers or suggestio