gpt-2 src/model.py gelu使用numpy函数

dsekswqp 于 6个月前发布在其他

关注(0)|答案(1)|浏览(100)

src/model.py脚本中的gelu函数使用了numpy.sqrt和numpy.pi，这对GPU性能有什么影响，它是否可以与GPU一起使用？如果不能，应该在tf中将其更改为类似的函数。

来源：https://github.com/openai/gpt-2/issues/228

1条答案

按热度按时间

你是说这部分：np.sqrt(2/np.pi)？我认为可以用计算一次的常数值（0.7978845608028654）来替换。只需决定精度就足够了。

赞(0）回复(0）举报 6个月前

相关问题

热门标签

Java query python Node 开发语言 request Util 数据库 Table 后端算法 Logger Message Element Parser

最新问答

xxl-job 安全组扫描到执行器端口服务存在信息泄露漏洞
回答(1) 发布于 4个月前
xxl-job 不能和nacos兼容？
回答(3) 发布于 4个月前
xxl-job 任务执行完后无法结束，日志一直转圈
回答(3) 发布于 4个月前
xxl-job-admin页面上查看调度日志样式问题
回答(1) 发布于 4个月前
xxl-job 参数512字符限制能否去掉
回答(1) 发布于 4个月前