!10098 fix bug of pynative's mixprecision

From: @lianliguang Reviewed-by: @kisnwang,@zhoufeng54 Signed-off-by: @zhoufeng54
2020-12-18 09:49:15 +08:00 · 2020-12-18 09:49:15 +08:00 · 885f41549a
parent c2579d1dc6 168fd6c9b2
commit 885f41549a
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/cell.py
+++ b/mindspore/nn/cell.py
@ -647,7 +647,7 @@ class Cell(Cell_):
                param.set_cast_dtype(mstype.float32)
            elif self._mindspore_flags.get('fp16'):
                param.set_cast_dtype(mstype.float16)
-            else:
+            elif hasattr(param, "set_cast_dtype"):
                # retest dtype
                param.set_cast_dtype()
        return param