!1710 Fix arg index check error when return Parameter defined in Net directly

Merge pull request !1710 from fary86/fix_arg_index_check_error_when_return_param
pull/1710/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 191dd59bc9

@ -422,8 +422,9 @@ bool IsGraphOutputValueNodeOrParameter(const AnfNodePtr &output, const py::tuple
MS_EXCEPTION(UnknownError) << "When graph output is Parameter, it should be found in graph parameters";
}
size_t index = it - params.cbegin();
if (index >= args.size()) {
MS_EXCEPTION(UnknownError) << "Index " << index << " equal or larger than args size " << args.size() << ".";
if (index >= args.size() + func_graph->hyper_param_count()) {
MS_EXCEPTION(UnknownError) << "Index " << index << " equal or larger than args size " << args.size()
<< " add Parameter count " << func_graph->hyper_param_count() << ".";
}
*ret_val = args[index];
return true;

Loading…
Cancel
Save