Skip to content

Commit a6fcf76

Browse files
committed
[layers] subpixel name scope
1 parent 5d01ad1 commit a6fcf76

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tensorlayer/layers.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2128,7 +2128,9 @@ def _PS(X, r, n_out_channel):
21282128
print(" [TL] SubpixelConv2d %s: scale: %d n_out_channel: %s act: %s" % (name, scale, n_out_channel, act.__name__))
21292129

21302130
net_new = Layer(inputs, name=name)
2131-
net_new.outputs = act(_PS(inputs, r=scale, n_out_channel=n_out_channel))
2131+
# with tf.name_scope(name):
2132+
with tf.variable_scope(name) as vs:
2133+
net_new.outputs = act(_PS(inputs, r=scale, n_out_channel=n_out_channel))
21322134

21332135
net_new.all_layers = list(net.all_layers)
21342136
net_new.all_params = list(net.all_params)

0 commit comments

Comments
 (0)