Skip to content

Commit 0392dec

Browse files
authored
Merge pull request #1079 from tiancheng2000/patch-1
in_channels must equal to previous n_filter
2 parents cf9392d + 50fd88a commit 0392dec

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorlayer/models/vgg.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def make_layers(config, batch_norm=False, end_with='outputs'):
123123
else:
124124
in_channels = 3
125125
else:
126-
in_channels = layer
126+
in_channels = layer_group[idx - 1]
127127
layer_list.append(
128128
Conv2d(
129129
n_filter=n_filter, filter_size=(3, 3), strides=(1, 1), act=tf.nn.relu, padding='SAME',

0 commit comments

Comments
 (0)