From 85bd04ba059c65174409f6fbe53d3ec3c8ae0f36 Mon Sep 17 00:00:00 2001 From: LianLiguang Date: Tue, 17 Nov 2020 09:47:12 +0800 Subject: [PATCH] fix bug of mixed precision --- mindspore/nn/cell.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/cell.py b/mindspore/nn/cell.py index 1e9e2364cc..9f5fda4de5 100755 --- 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)