79 lines
1.8 KiB
YAML
79 lines
1.8 KiB
YAML
mode: 'train'
|
|
ARCHITECTURE:
|
|
name: 'ResNet50_vd'
|
|
|
|
pretrained_model: "./pretrained/ResNet50_vd_pretrained"
|
|
model_save_dir: "./output/"
|
|
classes_num: 33
|
|
total_images: 17463
|
|
save_interval: 1
|
|
validate: True
|
|
valid_interval: 1
|
|
epochs: 10
|
|
topk: 1
|
|
image_shape: [3, 224, 224]
|
|
|
|
multilabel: True
|
|
|
|
use_mix: False
|
|
ls_epsilon: 0.1
|
|
|
|
LEARNING_RATE:
|
|
function: 'Cosine'
|
|
params:
|
|
lr: 0.07
|
|
|
|
OPTIMIZER:
|
|
function: 'Momentum'
|
|
params:
|
|
momentum: 0.9
|
|
regularizer:
|
|
function: 'L2'
|
|
factor: 0.000070
|
|
|
|
TRAIN:
|
|
batch_size: 256
|
|
num_workers: 4
|
|
file_list: "./dataset/NUS-WIDE-SCENE/NUS-SCENE-dataset/multilabel_train_list.txt"
|
|
data_dir: "./dataset/NUS-WIDE-SCENE/NUS-SCENE-dataset/images"
|
|
shuffle_seed: 0
|
|
transforms:
|
|
- DecodeImage:
|
|
to_rgb: True
|
|
to_np: False
|
|
channel_first: False
|
|
- RandCropImage:
|
|
size: 224
|
|
- RandFlipImage:
|
|
flip_code: 1
|
|
- NormalizeImage:
|
|
scale: 1./255.
|
|
mean: [0.485, 0.456, 0.406]
|
|
std: [0.229, 0.224, 0.225]
|
|
order: ''
|
|
- ToCHWImage:
|
|
mix:
|
|
- MixupOperator:
|
|
alpha: 0.2
|
|
|
|
VALID:
|
|
batch_size: 64
|
|
num_workers: 4
|
|
file_list: "./dataset/NUS-WIDE-SCENE/NUS-SCENE-dataset/multilabel_test_list.txt"
|
|
data_dir: "./dataset/NUS-WIDE-SCENE/NUS-SCENE-dataset/images"
|
|
shuffle_seed: 0
|
|
transforms:
|
|
- DecodeImage:
|
|
to_rgb: True
|
|
to_np: False
|
|
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: |