Update backend test (#990)

test/backend.py

@@ -10,7 +10,8 @@
 sys.pycache_prefix = os.path.join(root_dir, 'dist', 'pycache', 'test', 'backend')
 netron = __import__('source')
 
-test_data_dir = os.path.join(root_dir, 'third_party', 'test')
+third_party_dir = os.path.join(root_dir, 'third_party')
+test_data_dir = os.path.join(third_party_dir, 'test')
 
 def _test_onnx():
     file = os.path.join(test_data_dir, 'onnx', 'candy.onnx')
@@ -34,11 +35,18 @@ def _test_onnx_list():
 def _test_torchscript():
     torch = __import__('torch')
     torchvision = __import__('torchvision')
+    # model = torchvision.models.alexnet(weights=torchvision.models.AlexNet_Weights.DEFAULT)
     # model = torchvision.models.resnet34(weights=torchvision.models.ResNet34_Weights.DEFAULT)
-    model = torchvision.models.alexnet(weights=torchvision.models.AlexNet_Weights.DEFAULT)
+    model = torchvision.models.resnet34()
+    file = os.path.join(test_data_dir, 'pytorch', 'resnet34-333f7ec4.pth')
+    state_dict = torch.load(file)
+    model.load_state_dict(state_dict)
     args = torch.zeros([1, 3, 224, 224])
-    graph, _ = torch.jit._get_trace_graph(model, args) # pylint: disable=protected-access
-    # graph = torch.onnx._optimize_trace(graph, torch.onnx.OperatorExportTypes.ONNX)
+    # graph, _ = torch.jit._get_trace_graph(model, args) # pylint: disable=protected-access
+    # torch.onnx._optimize_trace(graph, torch.onnx.OperatorExportTypes.ONNX)
+    trace = torch.jit.trace(model, args, strict=True)
+    graph = trace.graph
+    torch._C._jit_pass_inline(graph)
     # https://github.com/pytorch/pytorch/blob/master/torch/csrc/jit/ir/ir.h
     netron.serve('resnet34', graph, browse=True, verbosity='quiet')