From f222ecfe5f09cb34b0a12b3a5c9f043c523be92e Mon Sep 17 00:00:00 2001 From: Eunju Yang Date: Wed, 2 Oct 2024 14:31:32 +0900 Subject: [PATCH] [ Print ] Update print result of model summary - This commit updates the model summary print of the layer with multiple inputs. [ASIS] concat0 concat 1:1:14:2 input0 1:1:4:2 input1 1:1:8:2 input2 [TOBE] concat0 concat 1:1:14:2 input0 input1 input2 Signed-off-by: Eunju Yang --- nntrainer/models/neuralnet.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/nntrainer/models/neuralnet.cpp b/nntrainer/models/neuralnet.cpp index d0e542825f..fd7dd7f032 100644 --- a/nntrainer/models/neuralnet.cpp +++ b/nntrainer/models/neuralnet.cpp @@ -1425,8 +1425,7 @@ void NeuralNetwork::print(std::ostream &out, unsigned int flags, out, {iter->getName(), iter->getType(), first_dim, first_input_name}); for (unsigned int i = 1; i < input_layer_names.size(); ++i) { dim_property.set(iter->getInputDimensions()[i]); - print_graph_layer_info( - out, {"", "", to_string(dim_property), input_layer_names[i]}); + print_graph_layer_info(out, {"", "", "", input_layer_names[i]}); } out << std::string(total_col_size, iter == model_graph.cend() - 1 ? '=' : '-')