File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -45,13 +45,13 @@ def _test_torchscript():
45
45
# graph, _ = torch.jit._get_trace_graph(model, args) # pylint: disable=protected-access
46
46
# torch.onnx._optimize_trace(graph, torch.onnx.OperatorExportTypes.ONNX)
47
47
# 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'))
48
50
# 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' ))
51
51
# trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'inception_v3_traced.pt'))
52
52
# 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'))
53
54
torch ._C ._jit_pass_inline (trace .graph )
54
- # https://github.com/pytorch/pytorch/blob/master/torch/csrc/jit/ir/ir.h
55
55
netron .serve ('resnet34' , trace )
56
56
57
57
# _test_onnx()
You can’t perform that action at this time.
0 commit comments