From 11f897ac2f1ca79c16a560104bd1b1c480ba2f87 Mon Sep 17 00:00:00 2001
From: Josh Pollock <joshpollock1997@gmail.com>
Date: Tue, 20 Nov 2018 22:50:37 -0800
Subject: [PATCH] fix dcgan layer naming overlap (#2145)
---
python/tvm/relay/testing/dcgan.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/relay/testing/dcgan.py b/python/tvm/relay/testing/dcgan.py
index 96cd871e4..d6c1d55df 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
--
GitLab