diff --git a/model_zoo/official/cv/resnet/resnet101_imagenet2012_config.yaml b/model_zoo/official/cv/resnet/resnet101_imagenet2012_config.yaml index d2727eb5bff..7a8ff11fe68 100644 --- a/model_zoo/official/cv/resnet/resnet101_imagenet2012_config.yaml +++ b/model_zoo/official/cv/resnet/resnet101_imagenet2012_config.yaml @@ -10,13 +10,13 @@ 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: '' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options -optimizer: 'Momentum' +optimizer: "Momentum" infer_label: "" class_num: 1001 batch_size: 32 @@ -60,15 +60,15 @@ 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' +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." diff --git a/model_zoo/official/cv/resnet/resnet18_cifar10_config.yaml b/model_zoo/official/cv/resnet/resnet18_cifar10_config.yaml index 1b6f7ba9e65..98708606c38 100644 --- a/model_zoo/official/cv/resnet/resnet18_cifar10_config.yaml +++ b/model_zoo/official/cv/resnet/resnet18_cifar10_config.yaml @@ -10,13 +10,13 @@ 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: '' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options -optimizer: 'Momentum' +optimizer: "Momentum" infer_label: "" class_num: 10 batch_size: 32 @@ -60,15 +60,15 @@ network_dataset: "resnet18_cifar10" --- # 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' +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." diff --git a/model_zoo/official/cv/resnet/resnet18_imagenet2012_config.yaml b/model_zoo/official/cv/resnet/resnet18_imagenet2012_config.yaml index b0834058f83..97a800959f1 100644 --- a/model_zoo/official/cv/resnet/resnet18_imagenet2012_config.yaml +++ b/model_zoo/official/cv/resnet/resnet18_imagenet2012_config.yaml @@ -10,13 +10,13 @@ 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: '' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options -optimizer: 'Momentum' +optimizer: "Momentum" infer_label: "" class_num: 1001 batch_size: 256 @@ -62,15 +62,15 @@ network_dataset: "resnet18_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' +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." diff --git a/model_zoo/official/cv/resnet/resnet34_imagenet2012_config.yaml b/model_zoo/official/cv/resnet/resnet34_imagenet2012_config.yaml index 41e029b2fa1..fe8dd627f58 100644 --- a/model_zoo/official/cv/resnet/resnet34_imagenet2012_config.yaml +++ b/model_zoo/official/cv/resnet/resnet34_imagenet2012_config.yaml @@ -10,13 +10,13 @@ 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: '' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options -optimizer: 'Momentum' +optimizer: "Momentum" infer_label: "" class_num: 1001 batch_size: 256 @@ -62,15 +62,15 @@ network_dataset: "resnet34_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' +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." diff --git a/model_zoo/official/cv/resnet/resnet50_cifar10_config.yaml b/model_zoo/official/cv/resnet/resnet50_cifar10_config.yaml index f0a1e5cdbd6..e568d5f0ef7 100644 --- a/model_zoo/official/cv/resnet/resnet50_cifar10_config.yaml +++ b/model_zoo/official/cv/resnet/resnet50_cifar10_config.yaml @@ -10,13 +10,13 @@ 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: '' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options -optimizer: 'Momentum' +optimizer: "Momentum" infer_label: "" class_num: 10 batch_size: 32 @@ -60,15 +60,15 @@ network_dataset: "resnet50_cifar10" --- # 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' +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." diff --git a/model_zoo/official/cv/resnet/resnet50_imagenet2012_Ascend_config.yaml b/model_zoo/official/cv/resnet/resnet50_imagenet2012_Ascend_config.yaml index b9a0037e978..643d0ca9b33 100644 --- a/model_zoo/official/cv/resnet/resnet50_imagenet2012_Ascend_config.yaml +++ b/model_zoo/official/cv/resnet/resnet50_imagenet2012_Ascend_config.yaml @@ -10,13 +10,13 @@ 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: '' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options -optimizer: 'Momentum' +optimizer: "Momentum" infer_label: "" class_num: 1001 batch_size: 32 @@ -63,15 +63,15 @@ network_dataset: "resnet50_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' +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." diff --git a/model_zoo/official/cv/resnet/resnet50_imagenet2012_GPU_config.yaml b/model_zoo/official/cv/resnet/resnet50_imagenet2012_GPU_config.yaml index 4fe6e61905a..603e4a9ae32 100644 --- a/model_zoo/official/cv/resnet/resnet50_imagenet2012_GPU_config.yaml +++ b/model_zoo/official/cv/resnet/resnet50_imagenet2012_GPU_config.yaml @@ -10,13 +10,13 @@ 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: '' +device_target: "GPU" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options -optimizer: 'Thor' +optimizer: "Thor" infer_label: "" class_num: 1001 batch_size: 32 @@ -63,15 +63,15 @@ network_dataset: "resnet50_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' +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." diff --git a/model_zoo/official/cv/resnet/resnet50_imagenet2012_config.yaml b/model_zoo/official/cv/resnet/resnet50_imagenet2012_config.yaml index b61f43c9caf..96c5566d448 100644 --- a/model_zoo/official/cv/resnet/resnet50_imagenet2012_config.yaml +++ b/model_zoo/official/cv/resnet/resnet50_imagenet2012_config.yaml @@ -10,13 +10,13 @@ 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: '' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options -optimizer: 'Momentum' +optimizer: "Momentum" infer_label: "" class_num: 1001 batch_size: 256 @@ -62,15 +62,15 @@ network_dataset: "resnet50_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' +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." diff --git a/model_zoo/official/cv/resnet/resnet_benchmark_GPU.yaml b/model_zoo/official/cv/resnet/resnet_benchmark_GPU.yaml index 157adf58cc4..c7c0455b4d6 100644 --- a/model_zoo/official/cv/resnet/resnet_benchmark_GPU.yaml +++ b/model_zoo/official/cv/resnet/resnet_benchmark_GPU.yaml @@ -10,13 +10,13 @@ enable_profiling: False data_path: "/cache/data" output_path: "/cache/train" load_path: "/cache/checkpoint_path/" -device_target: 'Ascend' -checkpoint_path: './checkpoint/' +device_target: "GPU" +checkpoint_path: "./checkpoint/" checkpoint_file_path: '' # ============================================================================== # Training options -optimizer: 'Momentum' +optimizer: "Momentum" infer_label: "" batch_size: 256 epoch_size: 2 @@ -37,15 +37,15 @@ network_dataset: "resnet50_cifar10" --- # 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' +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." diff --git a/model_zoo/official/cv/resnet/se-resnet50_imagenet2012_config.yaml b/model_zoo/official/cv/resnet/se-resnet50_imagenet2012_config.yaml index a6a45f06f53..a6e01b850cf 100644 --- a/model_zoo/official/cv/resnet/se-resnet50_imagenet2012_config.yaml +++ b/model_zoo/official/cv/resnet/se-resnet50_imagenet2012_config.yaml @@ -10,13 +10,13 @@ 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: '' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options -optimizer: 'Momentum' +optimizer: "Momentum" infer_label: "" class_num: 1001 batch_size: 32 @@ -54,8 +54,8 @@ mode_name: "GRAPH" # Export options device_id: 0 -width: 224 -height: 224 +width: 256 +height: 256 file_name: "se-resnet50" file_format: "AIR" ckpt_file: "" @@ -63,15 +63,15 @@ network_dataset: "se-resnet50_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' +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." diff --git a/model_zoo/official/cv/resnext/default_config.yaml b/model_zoo/official/cv/resnext/default_config.yaml index e17919c6c6a..bdd5d90e8b8 100644 --- a/model_zoo/official/cv/resnext/default_config.yaml +++ b/model_zoo/official/cv/resnext/default_config.yaml @@ -11,9 +11,9 @@ 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: '' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options @@ -22,7 +22,7 @@ num_classes: 1000 batch_size: 1 lr: 0.4 -lr_scheduler: 'cosine_annealing' +lr_scheduler: "cosine_annealing" lr_epochs: [30,60,90,120] lr_gamma: 0.1 eta_min: 0 @@ -43,7 +43,7 @@ ckpt_save_max: 5 is_save_on_master: 1 rank_save_ckpt_flag: 0 outputs_dir: "" -log_path: './output_log' +log_path: "./output_log" # Export options device_id: 0 @@ -56,17 +56,16 @@ label_path: "" --- # Help description for each configuration -network: "Candidate networks: [resnext50, resnext101], default: resnext50" -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/squeezenet/squeezenet_cifar10_config.yaml b/model_zoo/official/cv/squeezenet/squeezenet_cifar10_config.yaml index e34950541b5..5bc7187786f 100644 --- a/model_zoo/official/cv/squeezenet/squeezenet_cifar10_config.yaml +++ b/model_zoo/official/cv/squeezenet/squeezenet_cifar10_config.yaml @@ -12,9 +12,9 @@ output_path: "/cache/train" load_path: "/cache/checkpoint_path/" device_num: 1 device_id: 0 -device_target: 'Ascend' -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'suqeezenet_cifar10-120_195.ckpt' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "suqeezenet_cifar10-120_195.ckpt" # ============================================================================== # Training options @@ -45,16 +45,16 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/squeezenet/squeezenet_imagenet_config.yaml b/model_zoo/official/cv/squeezenet/squeezenet_imagenet_config.yaml index ce2935b39d4..1c2ed4f34da 100644 --- a/model_zoo/official/cv/squeezenet/squeezenet_imagenet_config.yaml +++ b/model_zoo/official/cv/squeezenet/squeezenet_imagenet_config.yaml @@ -12,9 +12,9 @@ output_path: "/cache/train" load_path: "/cache/checkpoint_path/" device_num: 1 device_id: 0 -device_target: 'Ascend' -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'suqeezenet_imagenet-200_5004.ckpt' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "suqeezenet_imagenet-200_5004.ckpt" # ============================================================================== # Training options diff --git a/model_zoo/official/cv/squeezenet/squeezenet_residual_cifar10_config.yaml b/model_zoo/official/cv/squeezenet/squeezenet_residual_cifar10_config.yaml index 906ffc35e89..2f00f6811dc 100644 --- a/model_zoo/official/cv/squeezenet/squeezenet_residual_cifar10_config.yaml +++ b/model_zoo/official/cv/squeezenet/squeezenet_residual_cifar10_config.yaml @@ -12,9 +12,9 @@ output_path: "/cache/train" load_path: "/cache/checkpoint_path/" device_num: 1 device_id: 0 -device_target: 'Ascend' -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'suqeezenet_residual_cifar10-150_195.ckpt' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "suqeezenet_residual_cifar10-150_195.ckpt" # ============================================================================== # Training options @@ -45,16 +45,16 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/squeezenet/squeezenet_residual_imagenet_config.yaml b/model_zoo/official/cv/squeezenet/squeezenet_residual_imagenet_config.yaml index 52943675e0d..9b4ea654da0 100644 --- a/model_zoo/official/cv/squeezenet/squeezenet_residual_imagenet_config.yaml +++ b/model_zoo/official/cv/squeezenet/squeezenet_residual_imagenet_config.yaml @@ -12,9 +12,9 @@ output_path: "/cache/train" load_path: "/cache/checkpoint_path/" device_num: 1 device_id: 0 -device_target: 'Ascend' -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'suqeezenet_residual_imagenet-300_5004.ckpt' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "suqeezenet_residual_imagenet-300_5004.ckpt" # ============================================================================== # Training options @@ -47,16 +47,16 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/ssd/ssd300_config.yaml b/model_zoo/official/cv/ssd/ssd300_config.yaml index 5520dc89b4a..a4ee378b9c0 100644 --- a/model_zoo/official/cv/ssd/ssd300_config.yaml +++ b/model_zoo/official/cv/ssd/ssd300_config.yaml @@ -10,9 +10,9 @@ 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: 'ssd-500_458.ckpt' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "ssd-500_458.ckpt" # ============================================================================== # Training options @@ -108,16 +108,16 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/ssd/ssd_mobilenet_v1_fpn_config.yaml b/model_zoo/official/cv/ssd/ssd_mobilenet_v1_fpn_config.yaml index d94a0fe4b10..6039cd4ccae 100644 --- a/model_zoo/official/cv/ssd/ssd_mobilenet_v1_fpn_config.yaml +++ b/model_zoo/official/cv/ssd/ssd_mobilenet_v1_fpn_config.yaml @@ -10,9 +10,9 @@ 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: 'mobilenet_v1.ckpt' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "mobilenet_v1.ckpt" # ============================================================================== # Training options diff --git a/model_zoo/official/cv/ssd/ssd_resnet50_fpn_config.yaml b/model_zoo/official/cv/ssd/ssd_resnet50_fpn_config.yaml index 6db091d4065..8b29de8779f 100644 --- a/model_zoo/official/cv/ssd/ssd_resnet50_fpn_config.yaml +++ b/model_zoo/official/cv/ssd/ssd_resnet50_fpn_config.yaml @@ -10,9 +10,9 @@ 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: 'resnet50.ckpt' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "resnet50.ckpt" # ============================================================================== # Training options @@ -113,16 +113,16 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/ssd/ssd_vgg16_config.yaml b/model_zoo/official/cv/ssd/ssd_vgg16_config.yaml index 28e1745aa9c..4704bfdcaa9 100644 --- a/model_zoo/official/cv/ssd/ssd_vgg16_config.yaml +++ b/model_zoo/official/cv/ssd/ssd_vgg16_config.yaml @@ -10,9 +10,9 @@ 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: 'ssd-500_458.ckpt' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "ssd-500_458.ckpt" # ============================================================================== # Training options @@ -107,16 +107,16 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/unet/unet_medical_config.yaml b/model_zoo/official/cv/unet/unet_medical_config.yaml index 033db4a28cc..bfaa3df493c 100644 --- a/model_zoo/official/cv/unet/unet_medical_config.yaml +++ b/model_zoo/official/cv/unet/unet_medical_config.yaml @@ -8,7 +8,7 @@ checkpoint_url: "" data_path: "/cache/data" output_path: "/cache/train" load_path: "/cache/checkpoint_path/" -device_target: 'Ascend' +device_target: "Ascend" enable_profiling: False # ============================================================================== @@ -30,17 +30,17 @@ weight_decay: 0.0005 loss_scale: 1024.0 FixedLossScaleManager: 1024.0 resume: False -resume_ckpt: './' +resume_ckpt: "./" transfer_training: False -filter_weight: ['outc.weight', 'outc.bias'] +filter_weight: ["outc.weight", "outc.bias"] #Eval options keep_checkpoint_max: 10 -eval_activate: 'Softmax' +eval_activate: "Softmax" eval_resize: False -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'ckpt_unet_medical_adam-4-75.ckpt' -rst_path: './result_Files/' +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "ckpt_unet_medical_adam-4-75.ckpt" +rst_path: "./result_Files/" result_path: "" # Export options @@ -51,16 +51,16 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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" weight_decay: "Weight decay." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/unet/unet_medical_gpu_config.yaml b/model_zoo/official/cv/unet/unet_medical_gpu_config.yaml index 48f2288315d..ec3e841203d 100644 --- a/model_zoo/official/cv/unet/unet_medical_gpu_config.yaml +++ b/model_zoo/official/cv/unet/unet_medical_gpu_config.yaml @@ -8,12 +8,12 @@ checkpoint_url: "" data_path: "/cache/data" output_path: "/cache/train" load_path: "/cache/checkpoint_path/" -device_target: 'GPU' +device_target: "GPU" enable_profiling: False # ============================================================================== # Training options -model_name: 'unet_medical' +model_name: "unet_medical" run_eval: False run_distribute: False crop: [388, 388] @@ -31,17 +31,17 @@ weight_decay: 0.0005 loss_scale: 1024.0 FixedLossScaleManager: 1024.0 resume: False -resume_ckpt: './' +resume_ckpt: "./" transfer_training: False -filter_weight: ['outc.weight', 'outc.bias'] +filter_weight: ["outc.weight", "outc.bias"] #Eval options keep_checkpoint_max: 10 -eval_activate: 'Softmax' +eval_activate: "Softmax" eval_resize: False -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'ckpt_unet_medical_adam-400.ckpt' -rst_path: './result_Files/' +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "ckpt_unet_medical_adam-400.ckpt" +rst_path: "./result_Files/" # Export options width: 572 @@ -51,16 +51,16 @@ file_format: AIR --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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" distribute_batchsize: "Batch size for distribute training" weight_decay: "Weight decay." diff --git a/model_zoo/official/cv/unet/unet_nested_cell_config.yaml b/model_zoo/official/cv/unet/unet_nested_cell_config.yaml index ca419d14887..6695a36c78c 100644 --- a/model_zoo/official/cv/unet/unet_nested_cell_config.yaml +++ b/model_zoo/official/cv/unet/unet_nested_cell_config.yaml @@ -8,15 +8,15 @@ checkpoint_url: "" data_path: "/cache/data" output_path: "/cache/train" load_path: "/cache/checkpoint_path/" -device_target: 'Ascend' +device_target: "Ascend" enable_profiling: False # ============================================================================== # Training options -model_name: 'unet_nested' +model_name: "unet_nested" run_eval: False run_distribute: False -dataset: 'Cell_nuclei' +dataset: "Cell_nuclei" crop: None image_size : [96, 96] lr: 0.0003 @@ -34,17 +34,17 @@ use_ds: False use_bn: False use_deconv: True resume: False -resume_ckpt: './' +resume_ckpt: "./" transfer_training: False -filter_weight: ['final1.weight', 'final2.weight', 'final3.weight', 'final4.weight'] +filter_weight: ["final1.weight", "final2.weight", "final3.weight", "final4.weight"] #Eval options keep_checkpoint_max: 10 -eval_activate: 'Softmax' +eval_activate: "Softmax" eval_resize: False -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'ckpt_unet_nested_adam-4-75.ckpt' -rst_path: './result_Files/' +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "ckpt_unet_nested_adam-4-75.ckpt" +rst_path: "./result_Files/" result_path: "./preprocess_Result" # Export options @@ -55,16 +55,16 @@ file_format: "MINDIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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" weight_decay: "Weight decay." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/unet/unet_nested_config.yaml b/model_zoo/official/cv/unet/unet_nested_config.yaml index 7357f8d937c..cf48c5c961b 100644 --- a/model_zoo/official/cv/unet/unet_nested_config.yaml +++ b/model_zoo/official/cv/unet/unet_nested_config.yaml @@ -8,12 +8,12 @@ checkpoint_url: "" data_path: "/cache/data" output_path: "/cache/train" load_path: "/cache/checkpoint_path/" -device_target: 'Ascend' +device_target: "Ascend" enable_profiling: False # ============================================================================== # Training options -model_name: 'unet_nested' +model_name: "unet_nested" run_eval: False run_distribute: False crop: None @@ -33,17 +33,17 @@ use_ds: True use_bn: True use_deconv: True resume: False -resume_ckpt: './' +resume_ckpt: "./" transfer_training: False -filter_weight: ['final1.weight', 'final2.weight', 'final3.weight', 'final4.weight'] +filter_weight: ["final1.weight", "final2.weight", "final3.weight", "final4.weight"] #Eval options keep_checkpoint_max: 10 -eval_activate: 'Softmax' +eval_activate: "Softmax" eval_resize: False -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'ckpt_unet_nested_adam-4-75.ckpt' -rst_path: './result_Files/' +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "ckpt_unet_nested_adam-4-75.ckpt" +rst_path: "/result_Files/" result_path: "" # Export options @@ -54,16 +54,16 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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" weight_decay: "Weight decay." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/unet/unet_simple_coco_config.yaml b/model_zoo/official/cv/unet/unet_simple_coco_config.yaml index 9c05ebf8ecb..1299dc88feb 100644 --- a/model_zoo/official/cv/unet/unet_simple_coco_config.yaml +++ b/model_zoo/official/cv/unet/unet_simple_coco_config.yaml @@ -8,15 +8,15 @@ checkpoint_url: "" data_path: "/cache/data" output_path: "/cache/train" load_path: "/cache/checkpoint_path/" -device_target: 'Ascend' +device_target: "Ascend" enable_profiling: False # ============================================================================== # Training options -model_name: 'unet_simple' +model_name: "unet_simple" run_eval: False run_distribute: False -dataset: 'COCO' +dataset: "COCO" split : 1.0 image_size : [512, 512] lr: 0.0001 @@ -31,7 +31,7 @@ weight_decay: 0.0005 loss_scale: 1024.0 FixedLossScaleManager: 1024.0 resume: False -resume_ckpt: './' +resume_ckpt: "./" transfer_training: False filter_weight: ['final1.weight'] coco_classes: ['background', 'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', @@ -60,11 +60,11 @@ eval_metrics: "dice_coeff" eval_start_epoch: 0 eval_interval: 1 keep_checkpoint_max: 10 -eval_activate: 'Softmax' +eval_activate: "Softmax" eval_resize: False -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'ckpt_unet_simple_adam-4-75.ckpt' -rst_path: './result_Files/' +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "ckpt_unet_simple_adam-4-75.ckpt" +rst_path: "./result_Files/" result_path: "" # Export options @@ -75,16 +75,16 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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" weight_decay: "Weight decay." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/unet/unet_simple_config.yaml b/model_zoo/official/cv/unet/unet_simple_config.yaml index 48dd8026eec..9e3238273fd 100644 --- a/model_zoo/official/cv/unet/unet_simple_config.yaml +++ b/model_zoo/official/cv/unet/unet_simple_config.yaml @@ -8,12 +8,12 @@ checkpoint_url: "" data_path: "/cache/data" output_path: "/cache/train" load_path: "/cache/checkpoint_path/" -device_target: 'Ascend' +device_target: "Ascend" enable_profiling: False # ============================================================================== # Training options -model_name: 'unet_simple' +model_name: "unet_simple" run_eval: False run_distribute: False crop: None @@ -30,17 +30,17 @@ weight_decay: 0.0005 loss_scale: 1024.0 FixedLossScaleManager: 1024.0 resume: False -resume_ckpt: './' +resume_ckpt: "./" transfer_training: False -filter_weight: ['final1.weight'] +filter_weight: ["final1.weight"] #Eval options keep_checkpoint_max: 10 -eval_activate: 'Softmax' +eval_activate: "Softmax" eval_resize: False -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'ckpt_unet_simple_adam-4-75.ckpt' -rst_path: './result_Files/' +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "ckpt_unet_simple_adam-4-75.ckpt" +rst_path: "./result_Files/" result_path: "./preprocess_Result" # Export options @@ -51,16 +51,16 @@ file_format: "MINDIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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" weight_decay: "Weight decay." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/cv/unet3d/default_config.yaml b/model_zoo/official/cv/unet3d/default_config.yaml index 45717e0f577..9292e77eb8b 100644 --- a/model_zoo/official/cv/unet3d/default_config.yaml +++ b/model_zoo/official/cv/unet3d/default_config.yaml @@ -11,9 +11,9 @@ 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: 'Unet3d-9-877.ckpt' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "Unet3d-9-877.ckpt" # ============================================================================== # Training options @@ -41,17 +41,17 @@ file_format: "" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -enable_fp16_gpu: 'Whether training on gpu with fp16, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +enable_fp16_gpu: "Whether training on gpu with fp16, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/nlp/mass/default_config.yaml b/model_zoo/official/nlp/mass/default_config.yaml index b24409434e6..86cdbc6236f 100644 --- a/model_zoo/official/nlp/mass/default_config.yaml +++ b/model_zoo/official/nlp/mass/default_config.yaml @@ -10,9 +10,9 @@ 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: '' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "" # ============================================================================== # Training options @@ -77,16 +77,16 @@ result_dir: "./result_Files" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint" diff --git a/model_zoo/official/nlp/textcnn/mr_config.yaml b/model_zoo/official/nlp/textcnn/mr_config.yaml index 9bfea2dc7f1..ff2f856c365 100644 --- a/model_zoo/official/nlp/textcnn/mr_config.yaml +++ b/model_zoo/official/nlp/textcnn/mr_config.yaml @@ -20,14 +20,14 @@ batch_size: 64 epoch_size: 4 weight_decay: 3e-5 keep_checkpoint_max: 1 -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'train_textcnn-4_149.ckpt' +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "train_textcnn-4_149.ckpt" word_len: 51 vec_length: 40 base_lr: 1e-3 label_dir: "" result_dir: "" -result_path: './preprocess_Result/' +result_path: "./preprocess_Result/" # Export options device_id: 0 @@ -36,15 +36,15 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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" dataset: "Dataset to be trained and evaluated, choice: ['MR, SUBJ, SST2']" train_epochs: "The number of epochs used to train." pre_trained: 'If need load pre_trained checkpoint, default: False' diff --git a/model_zoo/official/nlp/textcnn/mr_config_cpu.yaml b/model_zoo/official/nlp/textcnn/mr_config_cpu.yaml index f6ea63b96f1..896da5709b8 100644 --- a/model_zoo/official/nlp/textcnn/mr_config_cpu.yaml +++ b/model_zoo/official/nlp/textcnn/mr_config_cpu.yaml @@ -8,7 +8,7 @@ checkpoint_url: "" data_path: "./data" output_path: "./train" load_path: "./checkpoint_path/" -device_target: 'CPU' +device_target: "CPU" enable_profiling: False # ============================================================================== @@ -20,14 +20,14 @@ batch_size: 64 epoch_size: 4 weight_decay: 3e-5 keep_checkpoint_max: 1 -checkpoint_path: './checkpoint/' -checkpoint_file_path: './train/checkpoint/train_textcnn-4_149.ckpt' +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "./train/checkpoint/train_textcnn-4_149.ckpt" word_len: 51 vec_length: 40 base_lr: 1e-3 label_dir: "" result_dir: "" -result_path: './preprocess_Result/' +result_path: "./preprocess_Result/" # Export options device_id: 0 @@ -36,15 +36,15 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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" dataset: "Dataset to be trained and evaluated, choice: ['MR, SUBJ, SST2']" train_epochs: "The number of epochs used to train." pre_trained: 'If need load pre_trained checkpoint, default: False' diff --git a/model_zoo/official/nlp/textcnn/sst2_config.yaml b/model_zoo/official/nlp/textcnn/sst2_config.yaml index 01320c068de..1dbb8e15c7b 100644 --- a/model_zoo/official/nlp/textcnn/sst2_config.yaml +++ b/model_zoo/official/nlp/textcnn/sst2_config.yaml @@ -8,7 +8,7 @@ checkpoint_url: "" data_path: "/cache/data" output_path: "/cache/train" load_path: "/cache/checkpoint_path/" -device_target: 'Ascend' +device_target: "Ascend" enable_profiling: False # ============================================================================== @@ -20,14 +20,14 @@ batch_size: 64 epoch_size: 4 weight_decay: 3e-5 keep_checkpoint_max: 1 -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'train_textcnn-4_149.ckpt' +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "train_textcnn-4_149.ckpt" word_len: 51 vec_length: 40 base_lr: 5e-3 label_dir: "" result_dir: "" -result_path: './preprocess_Result/' +result_path: "/preprocess_Result/" # Export options device_id: 0 @@ -36,15 +36,15 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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" dataset: "Dataset to be trained and evaluated, choice: ['MR, SUBJ, SST2']" train_epochs: "The number of epochs used to train." pre_trained: 'If need load pre_trained checkpoint, default: False' diff --git a/model_zoo/official/nlp/textcnn/subj_config.yaml b/model_zoo/official/nlp/textcnn/subj_config.yaml index 3c2236fca2e..244b4928bdb 100644 --- a/model_zoo/official/nlp/textcnn/subj_config.yaml +++ b/model_zoo/official/nlp/textcnn/subj_config.yaml @@ -20,14 +20,14 @@ batch_size: 64 epoch_size: 4 weight_decay: 3e-5 keep_checkpoint_max: 1 -checkpoint_path: './checkpoint/' -checkpoint_file_path: 'train_textcnn-4_149.ckpt' +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "train_textcnn-4_149.ckpt" word_len: 51 vec_length: 40 base_lr: 8e-4 label_dir: "" result_dir: "" -result_path: './preprocess_Result/' +result_path: "./preprocess_Result/" # Export options device_id: 0 @@ -36,15 +36,15 @@ file_format: "AIR" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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" dataset: "Dataset to be trained and evaluated, choice: ['MR, SUBJ, SST2']" train_epochs: "The number of epochs used to train." pre_trained: 'If need load pre_trained checkpoint, default: False' diff --git a/model_zoo/research/cv/ssd_ghostnet/default_config.yaml b/model_zoo/research/cv/ssd_ghostnet/default_config.yaml index 1711ccc9aeb..936afb29741 100644 --- a/model_zoo/research/cv/ssd_ghostnet/default_config.yaml +++ b/model_zoo/research/cv/ssd_ghostnet/default_config.yaml @@ -10,9 +10,9 @@ 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: 'ssd-500_458.ckpt' +device_target: "Ascend" +checkpoint_path: "./checkpoint/" +checkpoint_file_path: "ssd-500_458.ckpt" # ============================================================================== # Training options @@ -93,16 +93,16 @@ image_dir: "" anno_path: "" --- # Help description for each configuration -enable_modelarts: 'Whether training on modelarts, default: False' -data_url: 'Dataset url for obs' -train_url: 'Training output 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' +enable_modelarts: "Whether training on modelarts, default: False" +data_url: "Dataset url for obs" +train_url: "Training output 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." keep_checkpoint_max: "keep the last keep_checkpoint_max checkpoint"