Is uses Pip to load a pre-trained model equal to use PyTorch Hub to load a pre-trained model?

rwightman / gen-efficientnet-pytorch

Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS

Apache License 2.0

1.56k stars 214 forks source link

Is uses Pip to load a pre-trained model equal to use PyTorch Hub to load a pre-trained model? #60

Closed TianhaoFu closed 3 years ago

TianhaoFu commented 3 years ago

I want use m = geffnet.create_model('mobilenetv3_large_100', pretrained=True) to replace model = torch.hub.load('rwightman/gen-efficientnet-pytorch', 'efficientnet_b0', pretrained=True) BTW, I use the pre-trained model aim to Fine-tuning the model.

Am i doing it right？

Thank you very much! @rwightman