未验证 提交 beef44b8 编写于 作者: F furnace 提交者: GitHub

bugfix for change arguments use_fp16 to use_amp, use_pure_fp16, multi_precision (#5012)

上级 6788ab2b
#SE_ResNet50_vd #SE_ResNet50_vd
export CUDA_VISIBLE_DEVICES=4 export CUDA_VISIBLE_DEVICES=0
export FLAGS_conv_workspace_size_limit=4000 #MB export FLAGS_conv_workspace_size_limit=4000 #MB
export FLAGS_cudnn_exhaustive_search=1 export FLAGS_cudnn_exhaustive_search=1
...@@ -9,7 +9,9 @@ export FLAGS_cudnn_batchnorm_spatial_persistent=1 ...@@ -9,7 +9,9 @@ export FLAGS_cudnn_batchnorm_spatial_persistent=1
DATA_DIR="Your image dataset path, e.g. /work/datasets/ILSVRC2012/" DATA_DIR="Your image dataset path, e.g. /work/datasets/ILSVRC2012/"
DATA_FORMAT="NHWC" DATA_FORMAT="NHWC"
USE_FP16=true #whether to use float16 USE_AMP=true #whether to use amp
USE_PURE_FP16=false
MULTI_PRECISION=${USE_PURE_FP16}
USE_DALI=true USE_DALI=true
USE_ADDTO=true USE_ADDTO=true
...@@ -26,7 +28,9 @@ python train.py \ ...@@ -26,7 +28,9 @@ python train.py \
--data_dir=${DATA_DIR} \ --data_dir=${DATA_DIR} \
--batch_size=128 \ --batch_size=128 \
--lr_strategy=cosine_decay \ --lr_strategy=cosine_decay \
--use_fp16=${USE_FP16} \ --use_amp=${USE_AMP} \
--use_pure_fp16=${USE_PURE_FP16} \
--multi_precision=${MULTI_PRECISION} \
--data_format=${DATA_FORMAT} \ --data_format=${DATA_FORMAT} \
--lr=0.1 \ --lr=0.1 \
--num_epochs=200 \ --num_epochs=200 \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册