Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Roshrini committed Dec 26, 2018
1 parent 3f03a17 commit 9b327b6
Showing 1 changed file with 0 additions and 22 deletions.
22 changes: 0 additions & 22 deletions tests/python-pytest/onnx/export/mxnet_export_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -394,28 +394,6 @@ def test_scalarops():
out = ipsym / 2
_test_scalar_op(input1, out, np_out)

def test_makeloss():
v1 = mx.nd.array([1, 2])
v2 = mx.nd.array([0, 1])
a = mx.sym.Variable('a')
b = mx.sym.Variable('b')
sym = mx.sym.MakeLoss(b + a)
ex = sym.bind(ctx=mx.cpu(0), args={'a': v1, 'b': v2})
ex.forward(is_train=True)
makeloss_out = ex.outputs[0].asnumpy()

converted_model = onnx_mxnet.export_model(sym, {}, [(2,),(2,)], np.float32, "makelossop.onnx")

# sym, arg_params, aux_params = onnx_mxnet.import_model(converted_model)
# result = forward_pass(sym, arg_params, aux_params, ['ipsym'], input1)
#
# # Comparing result of forward pass before using onnx export, import
# npt.assert_almost_equal(result, makeloss_out)
#
# executor = loss.simple_bind(ctx=mx.cpu(0), a=(2,), b=(2,))
# executor.forward(is_train=True, a=v1, b=v2)
# print(executor.outputs[0])

@with_seed()
def test_comparison_ops():
"""Test greater, lesser, equal"""
Expand Down

0 comments on commit 9b327b6

Please sign in to comment.