return super().find_class(mod_name, name)
AttributeError: Can't get attribute 'Identity' on <module 'efficientnet_pytorch.utils' from '/home/ai996/anaconda3/envs/pytorch/lib/python3.6/site-packages/efficientnet_pytorch/utils.py'>
efficientnet_pytorch版本太新,安装以前的版本。
pip install efficientnet_pytorch==0.6.0