From 00ff8e5ffddde035792e0fc695617452826d3f36 Mon Sep 17 00:00:00 2001 From: zhaoting Date: Wed, 9 Jun 2021 14:47:40 +0800 Subject: [PATCH] fix mobilenetv2 construct error --- model_zoo/official/cv/mobilenetv2/src/mobilenetV2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/cv/mobilenetv2/src/mobilenetV2.py b/model_zoo/official/cv/mobilenetv2/src/mobilenetV2.py index de4bdd672b5..a1fecc435b6 100644 --- a/model_zoo/official/cv/mobilenetv2/src/mobilenetV2.py +++ b/model_zoo/official/cv/mobilenetv2/src/mobilenetV2.py @@ -311,7 +311,7 @@ class MobileNetV2(nn.Cell): self.backbone = MobileNetV2Backbone(width_mult=width_mult, \ inverted_residual_setting=inverted_residual_setting, \ round_nearest=round_nearest, input_channel=input_channel, last_channel=last_channel).get_features - self.head = MobileNetV2Head(input_channel=self.backbone.out_channel, num_classes=num_classes, \ + self.head = MobileNetV2Head(input_channel=self.backbone.out_channels, num_classes=num_classes, \ has_dropout=has_dropout).get_head def construct(self, x):