Closed Bourn3z closed 5 months ago
Thank you for your contribution to the MindOCR repo. Before submitting this PR, please make sure:
为BaseModel类中的各模块设置amp_level,以便拿到属性并设置混合精度。 Set amp_level for each module in the BaseModel class to get attributes and set mixed precision.
(Write your motivation for proposed changes here.)
(How should this PR be tested? Do you require special setup to run the test or repro the fixed bug?)
(Is this PR part of a group of changes? Link the other relevant PRs and Issues here. Use https://help.github.com/en/articles/closing-issues-using-keywords for help on GitHub syntax)
Thank you for your contribution to the MindOCR repo. Before submitting this PR, please make sure:
Motivation
为BaseModel类中的各模块设置amp_level,以便拿到属性并设置混合精度。 Set amp_level for each module in the BaseModel class to get attributes and set mixed precision.
(Write your motivation for proposed changes here.)
Test Plan
(How should this PR be tested? Do you require special setup to run the test or repro the fixed bug?)
Related Issues and PRs
(Is this PR part of a group of changes? Link the other relevant PRs and Issues here. Use https://help.github.com/en/articles/closing-issues-using-keywords for help on GitHub syntax)