How can I change resnet18 to mobilenetv2?

[cuiwenhao]

No description provided.

[cuiwenhao]

I search many solutions, But I can not find a proper code to help me, by the way, I already find mobilenetv2.py, but I don't know how to use it in ESANet,Anyone can help me?

[mona0809]

You need to add another case here. But you also need to make sure to define some parameters in your MobileNetv2 implementation as well. However, I still think you need to rewrite the forward loop of the model as mobilenetv2 is built differently and does not have a max pool operation for example.