paddle 无法使用jit比jax训练慢

xlpyo6sf 于 5个月前发布在其他

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

需求描述 Feature Description

任务目标：复现 CausalPINN 代码

需求场景 每次迭代训练速度paddle比 jit() 下的jax慢三倍（0.1s vs 0.3s）

功能描述 在paddle上实现 jit() 函数

来源：https://github.com/PaddlePaddle/Paddle/issues/65111

1条答案

按热度按时间

paddle版本代码： https://github.com/DUCH714/PaddleScience/blob/develop/jointContribution/causalpinn/NS_paddle.py

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

相关问题

热门标签

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个月前