Python relay_inception_v3示例

示例#1

文件： test_graph.py 项目： shjwudp/training_results_v0.7

def test_inception_v3():
    shape = {"data": (1, 3, 299, 299)}
    mx_sym = model_zoo.mx_inception_v3()
    mod, _ = relay.frontend.from_mxnet(mx_sym, shape)
    relay_mod = model_zoo.relay_inception_v3()
    compare_graph(mod, relay_mod)

示例#2

文件： test_graph.py 项目： LANHUIYING/tvm

def test_inception_v3():
    shape = {"data": (1, 3, 299, 299)}
    mx_sym = model_zoo.mx_inception_v3()
    from_mx_sym, _ = relay.frontend.from_mxnet(mx_sym, shape)
    relay_sym = model_zoo.relay_inception_v3()
    compare_graph(from_mx_sym, relay_sym)