lukemelas / EfficientNet-PyTorch

A PyTorch implementation of EfficientNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

two " def from_pretrained " function ?

alicera opened this issue · 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.