Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
c20edc78
Unverified
Commit
c20edc78
authored
Mar 11, 2022
by
Double_V
Committed by
GitHub
Mar 11, 2022
Browse files
Merge pull request #5697 from mmglove/dygraph
fix run_benchmark_det.sh
parents
6d3bc571
1bf47c5a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
benchmark/run_benchmark_det.sh
benchmark/run_benchmark_det.sh
+2
-1
No files found.
benchmark/run_benchmark_det.sh
View file @
c20edc78
#!/usr/bin/env bash
set
-xe
# 运行示例:CUDA_VISIBLE_DEVICES=0 bash run_benchmark.sh ${run_mode} ${bs_item} ${fp_item} 500 ${model_mode}
# 参数说明
function
_set_params
(){
...
...
@@ -34,11 +33,13 @@ function _train(){
train_cmd
=
"python tools/train.py "
${
train_cmd
}
""
;;
mp
)
rm
-rf
./mylog
train_cmd
=
"python -m paddle.distributed.launch --log_dir=./mylog --gpus=
$CUDA_VISIBLE_DEVICES
tools/train.py
${
train_cmd
}
"
;;
*
)
echo
"choose run_mode(sp or mp)"
;
exit
1
;
esac
# 以下不用修改
echo
${
train_cmd
}
timeout
15m
${
train_cmd
}
>
${
log_file
}
2>&1
if
[
$?
-ne
0
]
;
then
echo
-e
"
${
model_name
}
, FAIL"
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment