17 lines
344 B
YAML
17 lines
344 B
YAML
DATALOADER:
|
|
TRAIN_X:
|
|
BATCH_SIZE: 200
|
|
TEST:
|
|
BATCH_SIZE: 200
|
|
NUM_WORKERS: 8
|
|
|
|
INPUT:
|
|
SIZE: (224, 224)
|
|
INTERPOLATION: "bicubic"
|
|
PIXEL_MEAN: [0.48145466, 0.4578275, 0.40821073]
|
|
PIXEL_STD: [0.26862954, 0.26130258, 0.27577711]
|
|
TRANSFORMS: ["random_resized_crop", "random_flip", "normalize"]
|
|
|
|
MODEL:
|
|
BACKBONE:
|
|
NAME: "RN50" |