I\'m using Tensorflow 2.0 with Python to train an image classifier (Mask RCNN). I\'m using the file model_main_tf2.py to train the model, and have a dataset of images for tr