diff --git a/CHANGELOG.md b/CHANGELOG.md index bcacdc66d..8f019b46e 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -78,6 +78,7 @@ To release a new version, please update the changelog as followed: ### Deprecated ### Fixed +- pretrained model VGG: fix logic of `in_channels` initialization (#1078) ### Removed @@ -85,6 +86,8 @@ To release a new version, please update the changelog as followed: ### Contributors +- @tiancheng2000(#1078) + ## [2.2.2] - 2020-04-26 TensorLayer 2.2.2 is a maintenance release. diff --git a/tensorlayer/models/vgg.py b/tensorlayer/models/vgg.py index 28fc0a8e0..11cab6753 100644 --- a/tensorlayer/models/vgg.py +++ b/tensorlayer/models/vgg.py @@ -123,7 +123,7 @@ def make_layers(config, batch_norm=False, end_with='outputs'): else: in_channels = 3 else: - in_channels = layer + in_channels = layer_group[idx - 1] layer_list.append( Conv2d( n_filter=n_filter, filter_size=(3, 3), strides=(1, 1), act=tf.nn.relu, padding='SAME',