forked from mindspore-Ecosystem/mindspore
49 lines
1.1 KiB
YAML
49 lines
1.1 KiB
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']
|