更新readme

README.md

@@ -34,7 +34,7 @@ pip3 install -r requirements.txt
 
 ```
 export HIP_VISIBLE_DEVICES=0
-python3 main_acc.py --batch-size=64 --arch=resnet50 -j 6 --epochs=90 --save-path=/path/to/{save_model_dir} /path/to/{ImageNet_pytorch_data_dir}/
+python3 train.py --batch-size=64 --arch=resnet50 -j 6 --epochs=90 --save-path=/path/to/{save_model_dir} /path/to/{ImageNet_pytorch_data_dir}/
 ```
 
 其中，'/path/to/{save_model_dir}' 修改为model 想要保存的路径，'/path/to/{ImageNet_pytorch_data_dir}/'修改为ImageNet数据集所在路径，后续混合精度训练与多卡训练脚本修改相同，不再说明。
@@ -43,7 +43,7 @@ python3 main_acc.py --batch-size=64 --arch=resnet50 -j 6 --epochs=90 --save-path
 
 ```
 export HIP_VISIBLE_DEVICES=0
-python3 main_acc.py --batch-size=64 --arch=resnet50 -j 6 --epochs=90 --amp --opt-level O1 --loss-scale=dynamic --save-path=/path/to/{save_model_dir} /path/to/{ImageNet_pytorch_data_dir}/
+python3 train.py --batch-size=64 --arch=resnet50 -j 6 --epochs=90 --amp --opt-level O1 --loss-scale=dynamic --save-path=/path/to/{save_model_dir} /path/to/{ImageNet_pytorch_data_dir}/
 ```
 
 ### 单机四卡测试（单精度）
@@ -63,4 +63,4 @@ mpirun --allow-run-as-root --bind-to none -np 4 scrips/single_process.sh localho
 
 # 参考
 
-https://github.com/pytorch/examples/tree/master/imagenet
\ No newline at end of file
+https://github.com/pytorch/examples/tree/master/imagenet