提交 27392cf0 编写于 作者: S Sansa 提交者: Liangliang He

Update validate.py (#317)

Simple print fix so it will work with py3 and removed spaces as requested as well.
上级 5dde6796
...@@ -224,10 +224,10 @@ def validate_onnx_model(platform, device_type, model_file, input_file, ...@@ -224,10 +224,10 @@ def validate_onnx_model(platform, device_type, model_file, input_file,
import onnx import onnx
if backend == "tensorflow": if backend == "tensorflow":
from onnx_tf.backend import prepare from onnx_tf.backend import prepare
print "valivate on onnx tensorflow backend." print("valivate on onnx tensorflow backend.")
elif backend == "caffe2" or backend == "pytorch": elif backend == "caffe2" or backend == "pytorch":
from caffe2.python.onnx.backend import prepare from caffe2.python.onnx.backend import prepare
print "valivate on onnx caffe2 backend." print("valivate on onnx caffe2 backend.")
else: else:
common.MaceLogger.error( common.MaceLogger.error(
VALIDATION_MODULE, VALIDATION_MODULE,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册