diff --git a/official/cv/resnet/golden_stick/pruner/scop/resnet50_imagenet2012_config.yaml b/official/cv/resnet/golden_stick/pruner/scop/resnet50_imagenet2012_config.yaml new file mode 100644 index 0000000000000000000000000000000000000000..325d9c39154d413af5369b23a08bd9056abd86fd --- /dev/null +++ b/official/cv/resnet/golden_stick/pruner/scop/resnet50_imagenet2012_config.yaml @@ -0,0 +1,109 @@ +# 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."