mixnet_s_ascend.yaml

# system
mode: 0
distribute: True
num_parallel_workers: 8
val_while_train: True

# dataset
dataset: "imagenet"
data_dir: "/path/to/imagenet"
shuffle: True
dataset_download: False
batch_size: 128
drop_remainder: True

# augmentation
image_resize: 224
scale: [0.08, 1.0]
ratio: [0.75, 1.333]
hflip: 0.5
interpolation: "bicubic"
auto_augment: "randaug-m9-mstd0.5"
re_prob: 0.25
crop_pct: 0.875
mixup: 0.2
cutmix: 1.0
cutmix_prob: 0.0

# model
model: "mixnet_s"
num_classes: 1000
pretrained: False
ckpt_path: ""
keep_checkpoint_max: 10
ckpt_save_dir: "./ckpt"
epoch_size: 600
dataset_sink_mode: True
amp_level: "O3"

# loss
loss: "CE"
label_smoothing: 0.1

# lr scheduler
scheduler: "cosine_decay"
lr: 0.2
min_lr: 0.00001
decay_epochs: 585
warmup_epochs: 15

# optimizer
opt: "momentum"
momentum: 0.9
weight_decay: 0.00002
loss_scale: 256
use_nesterov: False