-
Notifications
You must be signed in to change notification settings - Fork 1.1k
[Feat]: Add dynamic graph model pplcnet #4791
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
base: develop
Are you sure you want to change the base?
[Feat]: Add dynamic graph model pplcnet #4791
Conversation
|
Thanks for your contribution! |
| super().__init__(*args, **kwargs) | ||
| self.topk = topk | ||
| self.device = kwargs.get("device", None) | ||
| if is_bfloat16_available(self.device): |
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.
无需这段逻辑,后续没有用到self.dtype
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
| from .modeling import PPLCNet | ||
|
|
||
| if ( | ||
| self.model_name == "PP-LCNet_x1_0_doc_ori" |
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.
建议改成 self.model_name in xxx
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
No description provided.