!11655 [AutoParallel] log as INFO instead of EXCEPTION for unexpected output tensor shape cases

From: @ch-l
Reviewed-by: @zhoufeng54,@kisnwang
Signed-off-by: @kisnwang
pull/11655/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 6500655989

@ -487,7 +487,7 @@ Strategys MakeDataParallelStrategy(const std::shared_ptr<Graph> &graph,
} else if (ops[iter_ops]->outputs_tensor_info()[0].shape().size() == 4) {
graph->nodes[iter_graph].tensor_parm.tensor_str.str_n = 1.0 / std::min(max_device_num, target_tensor_batch);
} else {
MS_LOG(EXCEPTION) << ops[iter_ops]->name() << " output tensor shape is unexpected.";
MS_LOG(INFO) << ops[iter_ops]->name() << " output tensor shape is unexpected, using default value instead.";
}
return strategies;

Loading…
Cancel
Save