Skip to content

Commit 5738a6f

Browse files
authored
Merge pull request #278 from hysunflower/update_bs
[benchmark] update batch size for timesformer
2 parents e0da721 + 788c0f1 commit 5738a6f

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

benchmark/TimeSformer/run_all.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ cd ../../ # 返回PaddleVideo
3636

3737
model_mode_list=(TimeSformer)
3838
fp_item_list=(fp32 fp16)
39-
bs_item_list=(1 14)
39+
bs_item_list=(1) # 14
4040
for model_mode in ${model_mode_list[@]}; do
4141
for fp_item in ${fp_item_list[@]}; do
4242
for bs_item in ${bs_item_list[@]}

benchmark/TimeSformer/run_benchmark.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ function _set_params(){
1414
direction_id="0"
1515
ips_unit="instance/sec"
1616
skip_steps=1 # 解析日志,有些模型前几个step耗时长,需要跳过 (必填)
17-
keyword="ips:" # 解析日志,筛选出数据所在行的关键字 (必填)
17+
keyword="avg_ips:" # 解析日志,筛选出数据所在行的关键字 (必填)
1818
index="1"
1919
model_name=${model_item}_bs${batch_size}_${fp_item}
2020

0 commit comments

Comments
 (0)