AttributeError: Can‘t get attribute ‘Identity‘ on ＜module ‘efficientnet_pytorch.utils‘

环境：

• ubuntu16.04
• python3.7
• cuda == 10.1
• pytorch==1.6
• torchvision==0.7

在用efficientnet_pytorch包时，出现了如下报错：
Traceback (most recent call last):
File “model_test.py”, line 79, in
test(model_path,img_path,out_path)
File “model_test.py”, line 48, in test
model = torch.load(model_save_path)
File “/home/yu/anaconda3/envs/torch37/lib/python3.7/site-packages/torch/serialization.py”, line 584, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File “/home/yu/anaconda3/envs/torch37/lib/python3.7/site-packages/torch/serialization.py”, line 842, in _load
result = unpickler.load()
AttributeError: Can't get attribute 'Identity' on <module 'efficientnet_pytorch.utils' from ‘/home/yu/anaconda3/envs/torch37/lib/python3.7/site-packages/efficientnet_pytorch/utils.py’>

将efficientnet_pytorch 降为0.5版本时也报错：
AttributeError: Can't get attribute 'MemoryEfficientSwish' on <module ‘efficientnet_pytorch.utils’ from ‘/home/yu/anaconda3/envs/torch37/lib/python3.7/site-packages/efficientnet_pytorch/utils.py’>

解决办法：

pip install efficientnet_pytorch==0.6 

最后成功无报错！