Update auto_vot_train.py