提交 770a9723 编写于 作者: D dongshuilong

fix r50 benchmark static train yaml bug

上级 64033b36
...@@ -35,9 +35,9 @@ function _train(){ ...@@ -35,9 +35,9 @@ function _train(){
if [ ${fp_item} = "fp32" ]; then if [ ${fp_item} = "fp32" ]; then
config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50.yaml" config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50.yaml"
elif [ ${fp_item} = "amp_fp16" ]; then elif [ ${fp_item} = "amp_fp16" ]; then
config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1.yaml" config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1_ultra.yaml"
elif [ ${fp_item} = "pure_fp16" ]; then elif [ ${fp_item} = "pure_fp16" ]; then
config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2.yaml" config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2_ultra.yaml"
fi fi
if [ ${profiling} = "false" ]; then if [ ${profiling} = "false" ]; then
profiling_config="" profiling_config=""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册