コード例 #1
0
ファイル: test_graph.py プロジェクト: LANHUIYING/tvm 
def test_resnet():
    shape = {"data": (1, 3, 224, 224)}
    for n in [18, 34, 50, 101]:
        mx_sym = model_zoo.mx_resnet(n)
        from_mx_sym, _ = relay.frontend.from_mxnet(mx_sym, shape=shape)
        relay_sym = model_zoo.relay_resnet(n)
        compare_graph(from_mx_sym, relay_sym)
コード例 #2
0
def test_resnet():
    shape = {"data": (1, 3, 224, 224)}
    for n in [18, 34, 50, 101]:
        mx_sym = model_zoo.mx_resnet(n)
        mod, _ = relay.frontend.from_mxnet(mx_sym, shape=shape)
        relay_mod = model_zoo.relay_resnet(n)
        compare_graph(mod, relay_mod)
コード例 #3
0
ファイル: test_forward.py プロジェクト: mirsci/tvm 
def test_forward_resnet():
    for n in [18]:
        mx_sym = model_zoo.mx_resnet(18)
        verify_mxnet_frontend_impl(mx_sym)
コード例 #4
0
ファイル: test_forward.py プロジェクト: LANHUIYING/tvm 
def test_forward_resnet():
    for n in [18]:
        mx_sym = model_zoo.mx_resnet(18)
        verify_mxnet_frontend_impl(mx_sym)