fix bug of mixed precision

5 years ago · 85bd04ba05
parent c6c0a45c8a
commit 85bd04ba05
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/cell.py
+++ b/mindspore/nn/cell.py
@ -270,7 +270,7 @@ class Cell(Cell_):
                res.append(self._cast_mixed_precision_inputs(item, dst_type))
            elif isinstance(item, float):
                res.append(cast(item, dst_type))
-            elif hasattr(item, "dtype") in {mstype.float16, mstype.float32, mstype.float64}:
+            elif hasattr(item, "dtype") and item.dtype in {mstype.float16, mstype.float32, mstype.float64}:
                res.append(cast(item, dst_type))
            else:
                res.append(item)