lukemelas/EfficientNet-PyTorch

two " def from_pretrained " function ?

alicera opened this issue · 2 comments

in the efficientnet_pytorch/model.py , it has two function " from_pretrained "
what is the different?

def from_pretrained(cls, model_name, num_classes=1000, in_channels = 3):

def from_pretrained(cls, model_name, num_classes=1000):

Sorry about that! That's an error introduced in the last merge. I'll fix it in the next update to the repo.

Best,
Luke

Bug squashed.