mindspore/model_zoo/utils/model_scaffolding/config_imagenet.yaml

# Builtin Configurations(DO NOT CHANGE THESE CONFIGURATIONS unless you know exactly what you are doing)
enable_modelarts: False
# Url for modelarts
data_url: ""
train_url: ""
checkpoint_url: ""
# Path for local
data_path: "/cache/data"
output_path: "/cache/train"
load_path: "/cache/checkpoint_path"
device_target: Ascend
enable_profiling: False

# ==============================================================================
# Training options
epoch_size: 30
save_checkpoint: True
keep_checkpoint_max: 10
save_checkpoint_epochs: 5
save_checkpoint_steps: -1
ckpt_path: ""
lr: 0.1
momentum: 0.9
batch_size: 32
buffer_size: 1000
dataset_name: imagenet

# Model Description
model_name: lenet_running_8p
image_height: 32
image_width: 32
num_classes: 10


---
# Config description for each option
enable_modelarts: 'Whether training on modelarts, default: False'
data_url: 'Dataset url for obs'
train_url: 'Training output url for obs'
data_path: 'Dataset path for local'
output_path: 'Training output path for local'

device_target: 'Target device type'
enable_profiling: 'Whether enable profiling while training, default: False'


---
device_target: ['Ascend', 'GPU', 'CPU']