From 5fdd478b367fbe5e315e4eff70a15c31f44f84dc Mon Sep 17 00:00:00 2001 From: mmglove Date: Mon, 21 Mar 2022 21:07:33 +0800 Subject: [PATCH] fix static benchmark --- .../ResNet50/benchmark_common/run_benchmark.sh | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh index 4d5b2218..b420c739 100644 --- a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh +++ b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh @@ -45,7 +45,7 @@ function _train(){ log_file=${profiling_log_file} 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 DP) if [[ ${device_num} = "N1C1" ]];then @@ -76,12 +76,12 @@ function _train(){ } function _set_env(){ - #开启gc - export FLAGS_eager_delete_tensor_gb=0.0 - export FLAGS_fraction_of_gpu_memory_to_use=0.98 - #### + export FLAGS_fraction_of_gpu_memory_to_use=0.80 + export FLAGS_cudnn_batchnorm_spatial_persistent=1 + export FLAGS_max_inplace_grad_add=8 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 } -- GitLab