Skip to content
Snippets Groups Projects
Unverified Commit 2ac98f51 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!3950 add imagenet config for SCOP

Merge pull request !3950 from 185******25/master
parents 5f2dd09d 326bcdaf
No related branches found
No related tags found
No related merge requests found
# 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: 64
loss_scale: 1024
momentum: 0.9
weight_decay: 0.0001
epoch_size: 50
epochs_ft: 90
pretrain_epoch_size: 0
save_checkpoint: True
save_checkpoint_epochs: 5
keep_checkpoint_max: 10
warmup_epochs: 2
lr_decay_mode: "linear"
use_label_smooth: True
label_smooth_factor: 0.1
lr_init: 0.04
lr_end: 0.0
lr_max: 0.4
lars_epsilon: 0.0
lars_coefficient: 0.001
lr_ft_end: 0.0
lr_ft_max: 0.8
net_name: "resnet50"
dataset: "imagenet2012"
device_num: 1
ngpu: 1
pre_trained: ""
fp32_ckpt: ""
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"
boost_mode: "O0"
conv_init: "XavierUniform"
dense_init: "TruncatedNormal"
all_reduce_fusion_config:
- 85
- 160
train_image_size: 224
eval_image_size: 224
comp_algo: "scop"
#scop
prune_rate: 0.45
# Export options
device_id: 0
width: 224
height: 224
file_name: "resnet50"
file_format: "MINDIR"
ckpt_file: ""
network_dataset: "resnet50_imagenet"
# Retrain options
save_graphs: False
save_graphs_path: "./graphs"
has_trained_epoch: 0
has_trained_step: 0
# postprocess resnet inference
result_path: ''
label_path: ''
---
# 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."
save_graphs: "Whether save graphs during training, default: False."
save_graphs_path: "Path to save graphs."
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment