module 'torch.nn' has no attribute 'GELU'

File "/home/worker/anaconda3/envs/pytorch/lib/python3.7/site-packages/timm/models/layers/create_act.py", line 50, in <module>
gelu=nn.GELU,
AttributeError: module 'torch.nn' has no attribute 'GELU'

如果pytorch版本沒有問題可以直接修改vim /home/worker/anaconda3/envs/pytorch/lib/python3.7/site-packages/timm/models/layers/create_act.py

 

 

修改成

 

 解決

 

import torch.nn.functional as F