mindspore-lab / mindocr

A toolbox of ocr models and algorithms based on MindSpore
https://mindspore-lab.github.io/mindocr/
Apache License 2.0
206 stars 51 forks source link

Bugfix crnn #689

Closed Bourn3z closed 5 months ago

Bourn3z commented 5 months ago

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)