forked from mindspore-Ecosystem/mindspore
76 lines
2.0 KiB
YAML
76 lines
2.0 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
|
|
run_distribute: False
|
|
enable_profiling: False
|
|
data_path: "/cache/data"
|
|
output_path: "/cache/train"
|
|
load_path: "/cache/checkpoint_path/"
|
|
device_target: 'Ascend'
|
|
checkpoint_path: './checkpoint/'
|
|
checkpoint_file_path: ''
|
|
|
|
# ==============================================================================
|
|
# Training options
|
|
optimizer: 'Momentum'
|
|
infer_label: ""
|
|
class_num: 1001
|
|
batch_size: 32
|
|
loss_scale: 1024
|
|
momentum: 0.9
|
|
weight_decay: 0.0001
|
|
epoch_size: 120
|
|
pretrain_epoch_size: 0
|
|
save_checkpoint: True
|
|
save_checkpoint_epochs: 5
|
|
keep_checkpoint_max: 10
|
|
warmup_epochs: 0
|
|
lr_decay_mode: "cosine"
|
|
use_label_smooth: True
|
|
label_smooth_factor: 0.1
|
|
lr: 0.1
|
|
|
|
net_name: "resnet101"
|
|
dataset: "imagenet2012"
|
|
device_num: 1
|
|
pre_trained: ""
|
|
run_eval: False
|
|
eval_dataset_path: ""
|
|
parameter_server: False
|
|
filter_weight: False
|
|
save_best_ckpt: True
|
|
eval_start_epoch: 40
|
|
eval_interval: 1
|
|
enable_cache: False
|
|
cache_session_id: ""
|
|
mode_name: "GRAPH"
|
|
|
|
# Export options
|
|
device_id: 0
|
|
width: 224
|
|
height: 224
|
|
file_name: "resnet101"
|
|
file_format: "AIR"
|
|
ckpt_file: ""
|
|
network_dataset: "resnet101_imagenet2012"
|
|
|
|
---
|
|
# Help description for each configuration
|
|
enable_modelarts: 'Whether training on modelarts, default: False'
|
|
data_url: 'Dataset url for obs'
|
|
checkpoint_url: 'The location of checkpoint for obs'
|
|
data_path: 'Dataset path for local'
|
|
output_path: 'Training output path for local'
|
|
load_path: 'The location of checkpoint for obs'
|
|
device_target: 'Target device type, available: [Ascend, GPU, CPU]'
|
|
enable_profiling: 'Whether enable profiling while training, default: False'
|
|
num_classes: 'Class for dataset'
|
|
batch_size: "Batch size for training and evaluation"
|
|
epoch_size: "Total training epochs."
|
|
checkpoint_path: "The location of the checkpoint file."
|
|
checkpoint_file_path: "The location of the checkpoint file."
|