update readme

This commit is contained in:
l00486551 2021-08-05 11:09:58 +08:00
parent b5da59f2ac
commit ce28476a16
1 changed files with 8 additions and 4 deletions

View File

@ -86,12 +86,12 @@ WideResNet的总体网络架构如下[链接](https://arxiv.org/abs/1605.0714
# 分布式训练
用法:
cd scripts
bash run_distribute_train.sh [RANK_TABLE_FILE] [DATASET_PATH] [PRETRAINED_CKPT_PATH](可选) [MODELART]
bash run_distribute_train.sh [RANK_TABLE_FILE] [DATASET_PATH] [PRETRAINED_CKPT_PATH] [MODELART]
# 单机训练
用法:
cd scripts
bash run_standalone_train.sh [DATASET_PATH] [PRETRAINED_CKPT_PATH](可选) [MODELART]
bash run_standalone_train.sh [DATASET_PATH] [PRETRAINED_CKPT_PATH] [MODELART]
# 运行评估示例
用法:
@ -99,6 +99,8 @@ cd scripts
bash run_eval.sh [DATASET_PATH] [CHECKPOINT_PATH] [MODELART]
```
若没有[PRETRAINED_CKPT_PATH],使用 “” 作为参数运行脚本。
# 脚本说明
## 脚本及样例代码
@ -159,15 +161,17 @@ bash run_eval.sh [DATASET_PATH] [CHECKPOINT_PATH] [MODELART]
# 分布式训练
用法:
cd scripts
bash run_distribute_train.sh [RANK_TABLE_FILE] [DATASET_PATH] [PRETRAINED_CKPT_PATH](可选) [MODELART]
bash run_distribute_train.sh [RANK_TABLE_FILE] [DATASET_PATH] [PRETRAINED_CKPT_PATH] [MODELART]
# 单机训练
用法:
cd scripts
bash run_standalone_train.sh [DATASET_PATH] [PRETRAINED_CKPT_PATH](可选) [MODELART]
bash run_standalone_train.sh [DATASET_PATH] [PRETRAINED_CKPT_PATH] [MODELART]
```
若没有[PRETRAINED_CKPT_PATH],使用 “” 作为参数运行脚本。
分布式训练需要提前创建JSON格式的HCCL配置文件。
具体操作,参见[hccn_tools](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/utils/hccl_tools)中的说明。