diff --git a/python/tvm/relay/testing/dcgan.py b/python/tvm/relay/testing/dcgan.py index 96cd871e4122b55780f96f44023124560603d34c..d6c1d55df01a1354932beb56c765a1b17a308620 100644 --- a/python/tvm/relay/testing/dcgan.py +++ b/python/tvm/relay/testing/dcgan.py @@ -36,7 +36,7 @@ def deconv2d_bn_relu(data, prefix, **kwargs): """a block of deconv + batch norm + relu""" eps = 1e-5 + 1e-12 net = deconv2d(data, name="%s_deconv" % prefix, **kwargs) - net = layers.batch_norm_infer(net, epsilon=eps, name="batch_norm") + net = layers.batch_norm_infer(net, epsilon=eps, name="%s_batch_norm" % prefix) net = relay.nn.relu(net) return net