Skip to content

Commit 51ec793

Browse files
committed
Update backend test (#990)
1 parent 2414e06 commit 51ec793

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

Diff for: test/backend.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -45,13 +45,13 @@ def _test_torchscript():
4545
# graph, _ = torch.jit._get_trace_graph(model, args) # pylint: disable=protected-access
4646
# torch.onnx._optimize_trace(graph, torch.onnx.OperatorExportTypes.ONNX)
4747
# trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'fasterrcnn_resnet50_fpn.pt'))
48+
# torch.backends.quantized.engine = 'qnnpack'
49+
# trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'd2go.pt'))
4850
# trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'mobilenetv2-quant_full-nnapi.pt'))
49-
torch.backends.quantized.engine = 'qnnpack'
50-
trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'd2go.pt'))
5151
# trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'inception_v3_traced.pt'))
5252
# trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'netron_issue_920.pt'))
53+
# trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'bert-base-uncased.pt'))
5354
torch._C._jit_pass_inline(trace.graph)
54-
# https://github.com/pytorch/pytorch/blob/master/torch/csrc/jit/ir/ir.h
5555
netron.serve('resnet34', trace)
5656

5757
# _test_onnx()

0 commit comments

Comments
 (0)