two " def from_pretrained " function ?
alicera opened this issue · 2 comments
alicera commented
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):
lukemelas commented
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
lukemelas commented
Bug squashed.