EfficientNet-PyTorch icon indicating copy to clipboard operation
EfficientNet-PyTorch copied to clipboard

Published 20 hours ago • verified publisher icon lukemelas

from_pretrained() got an unexpected keyword argument 'weights_path'

Open onlinehuazai opened this issue 3 years ago • 1 comments

from_pretrained() got an unexpected keyword argument 'weights_path'

EfficientNet.from_pretrained(model_name='efficientnet-b3', weights_path='efficientnet-b3-5fb5a3c3.pth')

onlinehuazai avatar May 13 '22 11:05 onlinehuazai

Can you try like this?

model = EfficientNet.from_pretrained("efficientnet-b3", num_classes=2, weights_path="model.pth")

kadirnar avatar May 19 '22 13:05 kadirnar