未验证 提交 3e6150ba 编写于 作者: G gmm 提交者: GitHub

fix static benchmark (#5383)

* fix becnhmark,delete run_process_type

* fix

* fix benchmarkk

* fix mask_rcnn_r50_1x_coco benchmark

* fix benchmark
上级 c0b03467
...@@ -4,7 +4,7 @@ fp_item=fp32 ...@@ -4,7 +4,7 @@ fp_item=fp32
run_mode=DP run_mode=DP
device_num=N1C1 device_num=N1C1
max_iter=100 max_iter=100
num_workers=2 num_workers=8
# get data # get data
bash test_tipc/static/${model_item}/benchmark_common/prepare.sh bash test_tipc/static/${model_item}/benchmark_common/prepare.sh
......
...@@ -4,7 +4,7 @@ fp_item=fp32 ...@@ -4,7 +4,7 @@ fp_item=fp32
run_mode=DP run_mode=DP
device_num=N1C8 device_num=N1C8
max_iter=100 max_iter=100
num_workers=2 num_workers=8
# get data # get data
bash test_tipc/static/${model_item}/benchmark_common/prepare.sh bash test_tipc/static/${model_item}/benchmark_common/prepare.sh
......
...@@ -32,6 +32,7 @@ function _train(){ ...@@ -32,6 +32,7 @@ function _train(){
export FLAGS_eager_delete_tensor_gb=0.0 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.98
export FLAGS_memory_fraction_of_eager_deletion=1.0 export FLAGS_memory_fraction_of_eager_deletion=1.0
export FLAGS_conv_workspace_size_limit=500
cd ./static cd ./static
batch_size=${base_batch_size} # 如果模型跑多卡单进程时,请在_train函数中计算出多卡需要的bs batch_size=${base_batch_size} # 如果模型跑多卡单进程时,请在_train函数中计算出多卡需要的bs
echo "current CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES}, model_name=${model_name}, device_num=${device_num}, is profiling=${profiling}" echo "current CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES}, model_name=${model_name}, device_num=${device_num}, is profiling=${profiling}"
...@@ -49,8 +50,8 @@ function _train(){ ...@@ -49,8 +50,8 @@ function _train(){
use_fp16_cmd="" use_fp16_cmd=""
fi fi
train_cmd="-c configs/mask_rcnn_r50_1x.yml -o LearningRate.base_lr=0.001 snapshot_iter=100000 \ train_cmd="-c configs/mask_rcnn_r50_1x.yml \
TrainReader.batch_size=${batch_size} \ -o TrainReader.batch_size=${batch_size} \
max_iters=${max_iter} log_iter=1 \ max_iters=${max_iter} log_iter=1 \
TrainReader.worker_num=${num_workers} ${use_fp16_cmd} \ TrainReader.worker_num=${num_workers} ${use_fp16_cmd} \
--is_profiler=${is_profiler} " --is_profiler=${is_profiler} "
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册