mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

clear_warning

This commit is contained in:
fengyihang 2022-08-16 10:43:14 +08:00
parent 8ab98ed52f
commit 7e3572ce7a
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/common/tensor.py
View File

@ -5850,7 +5850,7 @@ def _check_astype_and_convert(dtype):
dtype = mstype.pytype_to_dtype(np.dtype(dtype.lower())) dtype = mstype.pytype_to_dtype(np.dtype(dtype.lower()))
elif isinstance(dtype, type): elif isinstance(dtype, type):
dtype = mstype.pytype_to_dtype(dtype) dtype = mstype.pytype_to_dtype(dtype)
elif not dtype in mstype.number_type + (mstype.bool_,): elif dtype not in mstype.number_type + (mstype.bool_,):
raise TypeError( raise TypeError(
f"For Tensor.astype, the input type must be one of {list(mstype.number_type + (mstype.bool_,) + np_types)}," f"For Tensor.astype, the input type must be one of {list(mstype.number_type + (mstype.bool_,) + np_types)},"
f" but got '{dtype}'.") f" but got '{dtype}'.")

4
mindspore/python/mindspore/train/amp.py
View File

@ -77,7 +77,7 @@ def _auto_white_list(network, white_list=None):
subcell = cells[name] subcell = cells[name]
if subcell == network: if subcell == network:
continue continue
elif isinstance(subcell, white_list): if isinstance(subcell, white_list):
network._cells[name] = _OutputTo32(subcell.to_float(mstype.float16)) network._cells[name] = _OutputTo32(subcell.to_float(mstype.float16))
change = True change = True
else: else:
@ -97,7 +97,7 @@ def _auto_black_list(network, black_list=None):
subcell = cells[name] subcell = cells[name]
if subcell == network: if subcell == network:
continue continue
elif isinstance(subcell, black_list): if isinstance(subcell, black_list):
network._cells[name] = _OutputTo16(subcell.to_float(mstype.float32)) network._cells[name] = _OutputTo16(subcell.to_float(mstype.float32))
change = True change = True
else: else: