Skip to content

Commit f864601

Browse files
committed
use benchmark configuration
1 parent f30ab00 commit f864601

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ascend_npu/torchbenchmark-config.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@ backend_extension: "torch_npu"
33
link: https://github.com/Ascend/pytorch
44
models:
55
- model: BERT_pytorch
6-
tests: [ "eval" ]
76
batch_size: 1
87

98
- model: yolov3
109
skip: true
10+
extra_args: "--accuracy"

0 commit comments

Comments
 (0)