Update auto_vot_train_after_fe.py