06_tpn_face.yaml

DATA:
  TRAIN_IMG_DATA: "image_data/train_data_face"
  VALID_IMG_DATA: "image_data/valid_data_face"
  TEST_IMG_DATA: "image_data/test_data_face"
DATA_LOADER:
  NAME: "tpn_data_loader"
  TRANSFORM: "set_tpn_transform_op"
  TRAIN_BATCH_SIZE: 2
MODEL:
  NAME: "tpn_model"
SOLVER:
  LR_INIT: 0.001
TRAIN:
  TRAINER: "TPNTrainer"
  OUTPUT_DIR: "results/unified_face_images/13_tpn"