问题:

运行程序报错:

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

方法:

import torch.nn.functional as F

把nn.RELU改为F.gelu

 

参考:

https://www.cnblogs.com/owlowl/p/13293391.html

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐