RuntimeError: Error(s) in loading state_dict for MobileNetV3:

[GHLab]

Thank you for your work.
Pull all source and create test.py and run it.

import torch
from mobilenetv3 import mobilenetv3_large, mobilenetv3_small

net_large = mobilenetv3_large()
net_small = mobilenetv3_small()

net_large.load_state_dict(torch.load('pretrained/mobilenetv3-large-1cd25616.pth'))
net_small.load_state_dict(torch.load('pretrained/mobilenetv3-small-55df8e1f.pth'))

But below error was occurred.

Traceback (most recent call last):
File "test.py", line 7, in
net_large.load_state_dict(torch.load('pretrained/mobilenetv3-large-1cd25616.pth'))
File "/usr/local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 839, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for MobileNetV3:
Missing key(s) in state_dict: "classifier.2.weight", "classifier.2.bias".
Unexpected key(s) in state_dict: "classifier.3.weight", "classifier.3.bias".

Could check this?

[Duo Li]

fixed