Exemplo n.º 1
0
def test_dcgan():
    shape = {"data": (2, 100)}
    mx_sym = model_zoo.mx_dcgan()
    mod, _ = relay.frontend.from_mxnet(mx_sym, shape)
    relay_mod = model_zoo.relay_dcgan(batch_size=2)
    compare_graph(mod, relay_mod)
Exemplo n.º 2
0
def test_dcgan():
    shape = {"data": (2, 100)}
    mx_sym = model_zoo.mx_dcgan()
    from_mx_sym, _ = relay.frontend.from_mxnet(mx_sym, shape)
    relay_sym = model_zoo.relay_dcgan(batch_size=2)
    compare_graph(from_mx_sym, relay_sym)