Repo for training and inferring BiT model Convert the model to ONNX format python -m tf2onnx.convert --saved-model /path/to/saved_model --output model.onnx