fix bug of pynative mixedprecision

4 years ago · 168fd6c9b2
parent 3f9e90a0a8
commit 168fd6c9b2
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/cell.py
+++ b/mindspore/nn/cell.py
@ -644,7 +644,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