p74368291
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!18052 fix mobilenetv2 construct error 

Merge pull request !18052 from zhaoting/master
This commit is contained in:
i-robot 2021-06-10 09:13:19 +08:00 committed by Gitee
parent 443510da86 00ff8e5ffd
commit 36ad5f85a0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/mobilenetv2/src/mobilenetV2.py
View File

@ -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):