diff --git a/dygraph/models/fcn.py b/dygraph/models/fcn.py index 5201cc98b090cc6e0fd48e093a8b43c277aca631..631e98fbbaad90936de49025883285b1eaf33955 100644 --- a/dygraph/models/fcn.py +++ b/dygraph/models/fcn.py @@ -70,11 +70,11 @@ class FCN(fluid.dygraph.Layer): self.model_pretrained = model_pretrained self.backbone_indices = backbone_indices if channels is None: - channels = backbone_channels[backbone_indices[0]] + channels = backbone_channels[0] self.backbone = backbone self.conv_last_2 = ConvBNLayer( - num_channels=backbone_channels[backbone_indices[0]], + num_channels=backbone_channels[0], num_filters=channels, filter_size=1, stride=1)