14 lines
225 B
YAML
14 lines
225 B
YAML
INPUT:
|
|
SIZE: (224, 224)
|
|
TRANSFORMS: ["random_flip", "random_translation", "normalize"]
|
|
|
|
DATASET:
|
|
NAME: "Office31"
|
|
|
|
MODEL:
|
|
BACKBONE:
|
|
NAME: "resnet50"
|
|
HEAD:
|
|
NAME: "mlp"
|
|
HIDDEN_LAYERS: [256]
|
|
DROPOUT: 0. |