from efficientnet_pytorch import EfficientNet
import torch

model = EfficientNet.from_name("efficientnet-b6")
from flyai.utils import remote_helper

path=remote_helper.get_remote_date("https://www.flyai.com/m/adv-efficientnet-b6-ac80338e.pth")

load_fc = False
state_dict = torch.load(path)
if load_fc:
    model.load_state_dict(state_dict)
else:
    state_dict.pop('_fc.weight')
    state_dict.pop('_fc.bias')
    res = model.load_state_dict(state_dict, strict=False)
    assert set(res.missing_keys) == set(['_fc.weight', '_fc.bias']), 'issue loading pretrained weights'