提交 5fdd478b 编写于 作者: M mmglove

fix static benchmark

上级 b51e58bf
...@@ -45,7 +45,7 @@ function _train(){ ...@@ -45,7 +45,7 @@ function _train(){
log_file=${profiling_log_file} log_file=${profiling_log_file}
fi fi
train_cmd="${config_file} -o DataLoader.Train.sampler.batch_size=${base_batch_size} -o Global.epochs=${max_epochs} -o DataLoader.Train.loader.num_workers=${num_workers} ${profiling_config} -o Global.eval_during_train=False" train_cmd="${config_file} -o DataLoader.Train.sampler.batch_size=${base_batch_size} -o Global.epochs=${max_epochs} -o DataLoader.Train.loader.num_workers=${num_workers} ${profiling_config} -o Global.eval_during_train=False -o fuse_elewise_add_act_ops=True -o enable_addto=True"
# 以下为通用执行命令,无特殊可不用修改 # 以下为通用执行命令,无特殊可不用修改
case ${run_mode} in case ${run_mode} in
DP) if [[ ${device_num} = "N1C1" ]];then DP) if [[ ${device_num} = "N1C1" ]];then
...@@ -76,12 +76,12 @@ function _train(){ ...@@ -76,12 +76,12 @@ function _train(){
} }
function _set_env(){ function _set_env(){
#开启gc export FLAGS_fraction_of_gpu_memory_to_use=0.80
export FLAGS_eager_delete_tensor_gb=0.0 export FLAGS_cudnn_batchnorm_spatial_persistent=1
export FLAGS_fraction_of_gpu_memory_to_use=0.98 export FLAGS_max_inplace_grad_add=8
####
export FLAGS_cudnn_exhaustive_search=1 export FLAGS_cudnn_exhaustive_search=1
export FLAGS_conv_workspace_size_limit=4000 #MB export FLAGS_eager_delete_tensor_gb=0.0
export FLAGS_conv_workspace_size_limit=4000
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册