eval.yaml 717 字节
Newer Older
W
WuHaobo 已提交
1
mode: 'valid'
W
WuHaobo 已提交
2
ARCHITECTURE:
S
shippingwang 已提交
3
    name: "ResNet50_vd"
W
WuHaobo 已提交
4

S
shippingwang 已提交
5
pretrained_model: "./pretrained_model/ResNet50_vd_pretrained"
W
WuHaobo 已提交
6 7 8 9 10 11 12 13
classes_num: 1000
total_images: 1281167
topk: 5
image_shape: [3, 224, 224]

VALID:
    batch_size: 16
    num_workers: 4
S
shippingwang 已提交
14 15
    file_list: "./dataset/ILSVRC2012/val_list.txt"
    data_dir: "./dataset/ILSVRC2012/"
W
WuHaobo 已提交
16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
    shuffle_seed: 0
    transforms:
        - DecodeImage:
            to_rgb: True
            channel_first: False
        - ResizeImage:
            resize_short: 256
        - CropImage:
            size: 224
        - NormalizeImage:
            scale: 1.0/255.0
            mean: [0.485, 0.456, 0.406]
            std: [0.229, 0.224, 0.225]
            order: ''
        - ToCHWImage: