Bugfix crnn (#689)

* Bugfix for crnn when amp_level is O0

* Bugfix for CRNN

Author: Bourn3z

mindocr/models/builder.py

@@ -7,7 +7,7 @@
 
 from ._registry import is_model, list_models, model_entrypoint
 from .base_model import BaseModel
-from .utils import load_model
+from .utils import load_model, set_amp_attr
 
 __all__ = ["build_model"]
 
@@ -74,5 +74,6 @@ def build_model(name_or_config: Union[str, dict], **kwargs):
 
     if "amp_level" in kwargs:
         auto_mixed_precision(network, amp_level=kwargs["amp_level"])
+        set_amp_attr(network, kwargs["amp_level"])
 
     return network

mindocr/models/utils/__init__.py

@@ -1,3 +1,3 @@
 from .attention_cells import *
-from .load_model import load_model
+from .load_model import load_model, set_amp_attr
 from .rnn_cells import GRUCell

mindocr/models/utils/load_model.py

@@ -2,11 +2,11 @@
 import os
 from typing import Callable, Dict, Optional
 
-from mindspore import load_checkpoint, load_param_into_net
+from mindspore import load_checkpoint, load_param_into_net, nn
 
 from ..backbones.mindcv_models.utils import auto_map, download_pretrained
 
-__all__ = ["load_model", "drop_inconsistent_shape_parameters"]
+__all__ = ["load_model", "drop_inconsistent_shape_parameters", "set_amp_attr"]
 _logger = logging.getLogger(__name__)
 
 
@@ -78,3 +78,9 @@ def load_model(
         f"Finish loading model checkoint from {load_from}. "
         "If no parameter fail-load warning displayed, all checkpoint params have been successfully loaded."
     )
+
+
+def set_amp_attr(network : nn.Cell, amp_level : str):
+    cells = network.name_cells()
+    for name in cells:
+        setattr(network._cells[name], "_amp_level", amp_level)