Example-Research-COAT/Code/Python/configs/prw-local.yaml

14 lines
247 B
YAML
Raw Normal View History

2024-10-03 01:04:42 +08:00
OUTPUT_DIR: "./logs/prw_coat"
INPUT:
DATASET: "PRW"
DATA_ROOT: "E:/DeepLearning/PersonSearch/COAT/datasets/PRW"
BATCH_SIZE_TRAIN: 3
SOLVER:
MAX_EPOCHS: 13
BASE_LR: 0.003
MODEL:
LOSS:
LUT_SIZE: 482
CQ_SIZE: 500
DISP_PERIOD: 100