sandeep-krishnamurthy commented on a change in pull request #10118: [MXNET-106][ONNX-MXNET] Adding ONNX Model zoo tests. URL: https://github.com/apache/incubator-mxnet/pull/10118#discussion_r177271724
########## File path: tests/python-pytest/onnx/onnx_test.py ########## @@ -21,19 +21,37 @@ ONNX backend test framework. Once we have PRs on the ONNX repo and get those PRs merged, this file will get EOL'ed. """ +# pylint: disable=too-many-locals,wrong-import-position,import-error from __future__ import absolute_import import sys import os import unittest import logging import hashlib +import tarfile +from collections import namedtuple import numpy as np import numpy.testing as npt from onnx import helper -import backend as mxnet_backend +from onnx import numpy_helper +from onnx import TensorProto +from mxnet.test_utils import download +from mxnet.contrib import onnx as onnx_mxnet +import mxnet as mx CURR_PATH = os.path.dirname(os.path.abspath(os.path.expanduser(__file__))) sys.path.insert(0, os.path.join(CURR_PATH, '../../python/unittest')) from common import with_seed +import backend as mxnet_backend + + +URLS = { + 'bvlc_googlenet' : Review comment: how big are these models? what is the impact on PR builds and nightly builds? ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services