-
Notifications
You must be signed in to change notification settings - Fork 30
Conversation
…TestBed get the code build fixes to ensure test_models pass.
@@ -49,7 +49,8 @@ def run_node(cls, node, inputs, device='CPU'): | |||
data_names = [i for i in node.input] | |||
data_shapes = [] | |||
reduce_op_types = set(['ReduceMin', 'ReduceMax', 'ReduceMean', |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me.
Change this variable name to something more relevant i.e dimChange_op_types or something other than "reduce"
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I will do the variable name change in one of the subsequent PRs.
@@ -263,5 +263,12 @@ def test_squeeze(self): | |||
output = mxnet_backend.run_node(node_def, [ip1])[0] | |||
npt.assert_almost_equal(output, np.squeeze(ip1, axis=[1, 3])) | |||
|
|||
def test_reshape(self): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
For tests like this - we'd encourage you to contribute to the main ONNX repo instead and run it through onnx backend test infrastructure. That way the tests would be shared across multiple backends. See https://github.com/onnx/onnx/blob/master/docs/OnnxBackendTest.md @lupesko @anirudhacharya
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes, we are in the process of doing that. We plan on switching our CI process from running these tests to running all the tests through the ONNX backend test infrastructure.
We are currently fixing a bunch of stuff on the import module so that all the operator conversions in our import module is completely in sync with the ONNX standards.
Soon after we plan to add support for more operators, and for each of those operators we will add corresponding tests on the ONNX repo.
Reshape already has tests in the ONNX repo. This PR in fact fixes the failures we were getting from running the very samee tests.
No description provided.