modify model_zoo resnet50 for R1.3

This commit is contained in:
lilei 2021-07-23 14:41:50 +08:00
parent 69a7ef8632
commit 5766636856
4 changed files with 8 additions and 6 deletions

View File

@ -230,9 +230,10 @@ If you want to run in modelarts, please check the official documentation of [mod
├── resnet18_imagenet2012_config.yaml # parameter configuration
├── resnet34_imagenet2012_config.yaml # parameter configuration
├── resnet50_cifar10_config.yaml # parameter configuration
├── resnet50_imagenet2012_Ascend_config.yaml # parameter configuration
├── resnet50_imagenet2012_Acc_config.yaml # parameter configuration
├── resnet50_imagenet2012_Ascend_Thor_config.yaml # parameter configuration
├── resnet50_imagenet2012_config.yaml # parameter configuration
├── resnet50_imagenet2012_GPU_config.yaml # parameter configuration
├── resnet50_imagenet2012_GPU_Thor_config.yaml # parameter configuration
├── resnet101_imagenet2012_config.yaml # parameter configuration
├── se-resnet50_imagenet2012_config.yaml # parameter configuration
├── export.py # export model for inference

View File

@ -213,9 +213,10 @@ bash run_eval_gpu.sh [DATASET_PATH] [CHECKPOINT_PATH] [CONFIG_PATH]
├── resnet18_imagenet2012_config.yaml # 参数配置
├── resnet34_imagenet2012_config.yaml # 参数配置
├── resnet50_cifar10_config.yaml # 参数配置
├── resnet50_imagenet2012_Ascend_config.yaml # 参数配置
├── resnet50_imagenet2012_Acc_config.yaml # 参数配置
├── resnet50_imagenet2012_Ascend_Thor_config.yaml # 参数配置
├── resnet50_imagenet2012_config.yaml # 参数配置
├── resnet50_imagenet2012_GPU_config.yaml # 参数配置
├── resnet50_imagenet2012_GPU_Thor_config.yaml # 参数配置
├── resnet101_imagenet2012_config.yaml # 参数配置
├── se-resnet50_imagenet2012_config.yaml # 参数配置
├── eval.py # 评估网络

View File

@ -16,7 +16,7 @@ checkpoint_file_path: ""
# ==============================================================================
# Training options
optimizer: "Momentum"
optimizer: "Thor"
infer_label: ""
class_num: 1001
batch_size: 32

View File

@ -16,7 +16,7 @@ checkpoint_file_path: ""
# ==============================================================================
# Training options
optimizer: "Momentum"
optimizer: "Thor"
infer_label: ""
class_num: 1001
batch_size: 32