modify model_zoo network bug
This commit is contained in:
parent
0022d07d6e
commit
f23cf7ac43
|
@ -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."
|
||||
|
|
|
@ -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."
|
||||
|
|
|
@ -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."
|
||||
|
|
|
@ -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."
|
||||
|
|
|
@ -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."
|
||||
|
|
|
@ -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."
|
||||
|
|
|
@ -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."
|
||||
|
|
|
@ -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."
|
||||
|
|
|
@ -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."
|
||||
|
|
|
@ -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."
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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."
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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'
|
||||
|
|
|
@ -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'
|
||||
|
|
|
@ -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'
|
||||
|
|
|
@ -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'
|
||||
|
|
|
@ -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"
|
||||
|
|
Loading…
Reference in New Issue