For simplicity, mobilenet_v2 from torchvision is used here.
from torchvision.models.mobilenet import mobilenet_v2
model = mobilenet_v2(pretrained=True)For simplicity, mobilenet_v2 from torchvision is used here.
from torchvision.models.mobilenet import mobilenet_v2
model = mobilenet_v2(pretrained=True)