From bd40bcd1a2e7ea7506bd4f1a262f7dce6f94b38b Mon Sep 17 00:00:00 2001
From: Wenhao Hu <fumihwh@gmail.com>
Date: Wed, 28 Mar 2018 13:26:22 +0900
Subject: [PATCH] use caffe2.python.onnx.backend instead of onnx-caffe2 (#418)
---
nnvm/tests/python/frontend/onnx/test_forward.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/nnvm/tests/python/frontend/onnx/test_forward.py b/nnvm/tests/python/frontend/onnx/test_forward.py
index 20fbcefad..d71e17183 100644
--- a/nnvm/tests/python/frontend/onnx/test_forward.py
+++ b/nnvm/tests/python/frontend/onnx/test_forward.py
@@ -7,9 +7,9 @@ import onnx
from model_zoo import super_resolution, squeezenet1_1, lenet, resnet18_1_0
def verify_onnx_forward_impl(graph_file, data_shape, out_shape):
- import onnx_caffe2.backend
+ import caffe2.python.onnx.backend
def get_caffe2_output(model, x, dtype='float32'):
- prepared_backend = onnx_caffe2.backend.prepare(model)
+ prepared_backend = caffe2.python.onnx.backend.prepare(model)
W = {model.graph.input[0].name: x.astype(dtype)}
c2_out = prepared_backend.run(W)[0]
return c2_out
--
GitLab