I am trying to run model_main_tf2.py to train models from detection zoo on custom dataset. I am training it on Kaggle GPU machine with the tfrecord in the working directory. I a