[ Print ] Update print result of model summary
authorEunju Yang <ej.yang@samsung.com>
Wed, 2 Oct 2024 05:31:32 +0000 (14:31 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Fri, 4 Oct 2024 08:26:40 +0000 (17:26 +0900)
- 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 <ej.yang@samsung.com>
nntrainer/models/neuralnet.cpp

index d0e542825fa736265938cb0ae042c07fd3415227..fd7dd7f032d5ddef3eb8ed0ba377697c076d6640 100644 (file)
@@ -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 ? '=' : '-')