xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix_set_inputs_problem

This commit is contained in:
luochao 2023-01-17 16:45:11 +08:00
parent 2d968044fc
commit ce5241dda8
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/common/api.py
View File

@ -410,7 +410,7 @@ class _MindsporeFunctionExecutor:
# Case: If the shape of input args is dynamic, get dynamic shape tensor from context and use it to compile.
compile_args = args_list
# Case: The `set_inputs()` of Cell object has been set, using these dynamic shape args as compile args.
if isinstance(self.obj, ms.nn.Cell) and self.obj.get_inputs():
if self.fn.__name__ == 'construct' and isinstance(self.obj, ms.nn.Cell) and self.obj.get_inputs():
compile_args = self.obj.get_inputs()
for args in compile_args:
Validator.check_isinstance("args set in `set_inputs()` of Cell", args, PythonTensor)