DATASET:
     NAME: 'python'
     SELECT_CATS_LIST: [2]
     SUFFIX: 'heads'
TEST:
    MODEL_FILE: 'wbia_orientation/output/python_orientation_v0/best.pth'
    BS: 256
VERSION: v0
