Ejemplos de mx_squeezenet en Python

Ejemplo n.º 1

Archivo: test_graph.py Proyecto: shjwudp/training_results_v0.7

def test_squeezenet():
    shape = {"data": (1, 3, 224, 224)}
    for version in ['1.0', '1.1']:
        mx_sym = model_zoo.mx_squeezenet(version)
        mod, _ = relay.frontend.from_mxnet(mx_sym, shape)
        relay_mod = model_zoo.relay_squeezenet(version)
        compare_graph(mod, relay_mod)

Ejemplo n.º 2

Archivo: test_graph.py Proyecto: LANHUIYING/tvm

def test_squeezenet():
    shape = {"data": (1, 3, 224, 224)}
    for version in ['1.0', '1.1']:
        mx_sym = model_zoo.mx_squeezenet(version)
        from_mx_sym, _ = relay.frontend.from_mxnet(mx_sym, shape)
        relay_sym = model_zoo.relay_squeezenet(version)
        compare_graph(from_mx_sym, relay_sym)