Can‘t get attribute ‘SiLU‘ on <module ‘torch.nn.modules.activation
ubuntu 18.04 server
python3.8.0
pytorch==1.6.0 torchvision==0.7.0
解决方法
将SiLU拷贝到~/.local/lib/python3.8/site-packages/torch/nn/modules/activation.py文件中
class SiLU(Module): # export-friendly version of nn.SiLU()
@staticmethod
def forward(x):
return x * torch.sigmoid(x)