-
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.10】 cfdgcn #1037
base: develop
Are you sure you want to change the base?
Conversation
Thanks for your contribution! |
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.
这个pre-commit应该不用改吧
@@ -53,4 +53,4 @@ repos: | |||
files: \.(c|cc|cxx|cpp|cu|h|hpp|hxx|cuh|proto)$ | |||
|
|||
exclude: | | |||
^jointContribution/ | |||
^jointContribution/ |
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.
末尾换行不要删除
``` sh | ||
python cfdgcn.py mode=export | ||
``` | ||
|
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.
``` sh | ||
python cfdgcn.py mode=infer | ||
``` | ||
|
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.
input_spec = [ | ||
{key: InputSpec([None, 1], "float32", name=key) for key in model.input_keys}, | ||
] |
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.
cfdgcn的输入是[None, 1]的形状吗?感觉这里不太对
|
||
# 初始化预测器 |
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.
换行保留,但是中文注释可以删除
PR types
Others
PR changes
Others
Describe
为cfdgcn算子添加export和inference功能