Thank you! I found source code of MobileNetV3Small and
just removed some layer, it also works well.
However, I can’t use pretrained network and it has overfitting problem in the validation data set…
==> As you mentioned, when loading the pretrained model at load_state_dict,
using strict=False removed unwanted layer as Magic!!! Thank you!!! Pytorch is very Nice tool!!
Thank you!!! Thank you for saving my life!!! Thank you!!!