p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix bug of error log of interpolate

This commit is contained in:
mengyuanli 2023-01-05 09:51:59 +08:00
parent bd2141ce00
commit 5ce343cc02
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/ops/function/nn_func.py
View File

@ -1902,7 +1902,7 @@ def _check_interpolate_inputs(input_dims, roi, scales, sizes, coordinate_transfo
support_coordinate_mode_list = ["align_corners", "half_pixel", "asymmetric"] support_coordinate_mode_list = ["align_corners", "half_pixel", "asymmetric"]
if coordinate_transformation_mode not in support_coordinate_mode_list: if coordinate_transformation_mode not in support_coordinate_mode_list:
raise TypeError(f"{msg_prefix} coordinate_transformation_mode must be in {support_coordinate_mode_list}," raise TypeError(f"{msg_prefix} coordinate_transformation_mode must be in {support_coordinate_mode_list},"
" but got {coordinate_transformation_mode}") f" but got {coordinate_transformation_mode}")
validator.check_value_type("mode", mode, [str], prim_name) validator.check_value_type("mode", mode, [str], prim_name)
if mode == "linear": if mode == "linear":
validator.check_int(input_dims, 3, Rel.EQ, "input dims", prim_name) validator.check_int(input_dims, 3, Rel.EQ, "input dims", prim_name)