Add export & inference for hPINNs #902
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
… into dev_model feat: add export and infer functions for hpinns
|
Thanks for your contribution! |
lijialin03
reviewed
May 16, 2024
| model_re = ppsci.arch.MLP(**cfg.MODEL.re_net) | ||
| model_im = ppsci.arch.MLP(**cfg.MODEL.im_net) | ||
| model_eps = ppsci.arch.MLP(**cfg.MODEL.eps_net) | ||
| # wrap to a model_list |
There was a problem hiding this comment.
当前导出后推理会报错：
RuntimeError: (PreconditionNotMet) The variable named x is not found in the scope of the executor.
[Hint: scope->FindVar(name) should not be null.] (at ../paddle/fluid/inference/api/analysis_predictor.cc:2333)
这是由于导出时的input_keys和推理时的input_keys不匹配导致的。
这个模型中有input transform，即：("x","y")->transform->("x_cos_1","y_cons_1",...)->forward->("e_re",...)，当前export函数中的input_keys是("x_cos_1","y_cons_1",...)，而infer中是("x","y")。
因此，需要更改：
- 这里增加transform register代码
# register transform
model_re.register_input_transform(func_module.transform_in)
model_im.register_input_transform(func_module.transform_in)
model_eps.register_input_transform(func_module.transform_in)
model_re.register_output_transform(func_module.transform_out_real_part)
model_im.register_output_transform(func_module.transform_out_imaginary_part)
model_eps.register_output_transform(func_module.transform_out_epsilon)
- 更改下面
input_spec
lijialin03
reviewed
May 16, 2024
examples/hpinns/holography.py
Outdated
| input_spec = [ | ||
| { | ||
| key: InputSpec([None, 1], "float32", name=key) | ||
| for key in model_list.input_keys |
lijialin03
reviewed
May 16, 2024
| } | ||
|
|
||
| ppsci.visualize.save_vtu_from_dict( | ||
| "./hpinns_pred.vtu", |
There was a problem hiding this comment.
改为：osp.join(cfg.output_dir, "hpinns_pred.vtu"),以防文件覆盖
There was a problem hiding this comment.
改为：
osp.join(cfg.output_dir, "hpinns_pred.vtu"),以防文件覆盖
inference的产物文件就不放到output_dir下面了吧，这个就保持hpinns_pred.vtu比较好
… into dev_model fix:register transform before export
huohuohuohuohuo123
pushed a commit
to huohuohuohuohuo123/PaddleScience
that referenced
this pull request
Aug 12, 2024
* feat: add export and infer functions for hpinns * fix:register transform brfore export
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
Others
PR changes
APIs
Describe
为hPINNs添加导出和推理代码