Commit 585bd1ca by Tatsuya Nishiyama Committed by Tianqi Chen

Add dtype option to verify_mxnet_frontend_impl (#1908)

parent f01cc0e6
......@@ -14,7 +14,7 @@ import model_zoo
def verify_mxnet_frontend_impl(mx_symbol, data_shape=(1, 3, 224, 224), out_shape=(1, 1000),
gluon_impl=False, name=None):
gluon_impl=False, name=None, dtype='float32'):
"""Use name different from test to avoid let nose pick it up"""
if gluon_impl:
def get_gluon_output(name, x):
......@@ -57,7 +57,6 @@ def verify_mxnet_frontend_impl(mx_symbol, data_shape=(1, 3, 224, 224), out_shape
return out.asnumpy()
# random input
dtype = 'float32'
x = np.random.uniform(size=data_shape)
if gluon_impl:
gluon_out, gluon_sym = get_gluon_output(name, x)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment