Update voc-trainer.py