-
Notifications
You must be signed in to change notification settings - Fork 184
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
【SCU】【PPSCI Export&Infer No.17】poiseuille_flow #1030
Conversation
Thanks for your contribution! |
@HydrogenSulfate 拜托佬再看看!感谢~ |
model_u = ppsci.arch.MLP(**cfg.MODEL.u_net) | ||
model_v = ppsci.arch.MLP(**cfg.MODEL.v_net) | ||
model_p = ppsci.arch.MLP(**cfg.MODEL.p_net) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
- 看起来缺少这部分代码代码,image
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done. 之前测试可以导出就没加这部分代码,已经修改好了。
Co-authored-by: HydrogenSulfate <490868991@qq.com>
solver.export(input_spec, export_path) | ||
|
||
|
||
def inference(cfg: DictConfig): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
inference应该使用Predictor,加载export导出的模型进行计算,而不是使用solver
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
PR types
Others
PR changes
Others
Describe
#788
为pipe案例添加了export和inference功能