Caffe2 - Python API
A deep learning, cross platform ML framework
test_onnx_common.py
1 from __future__ import absolute_import
2 from __future__ import division
3 from __future__ import print_function
4 from __future__ import unicode_literals
5 
6 import os
7 
8 
9 onnx_model_dir = os.path.join(os.path.dirname(
10  os.path.realpath(__file__)), os.pardir, "repos", "onnx", "onnx",
11  "backend", "test", "data")
12 
13 
14 pytorch_converted_dir = os.path.join(onnx_model_dir, "pytorch-converted")
15 
16 
17 pytorch_operator_dir = os.path.join(onnx_model_dir, "pytorch-operator")