Commit 11f897ac by Josh Pollock Committed by Lianmin Zheng

fix dcgan layer naming overlap (#2145)

parent f373d6a0
......@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment