DATA:
  IMG_SIZE: 128
  BATCH_SIZE: 1
  NUM_WORKERS: 1
MODEL:
  BACKBONE: resnet50
  EMB_SIZE: 512