11 lines
177 B
YAML
11 lines
177 B
YAML
INPUT:
|
|
SIZE: (224, 224)
|
|
TRANSFORMS: ["random_flip", "random_translation", "normalize"]
|
|
|
|
DATASET:
|
|
NAME: "PACS"
|
|
|
|
MODEL:
|
|
BACKBONE:
|
|
NAME: "resnet18"
|
|
PRETRAINED: True |