Skip to content

Commit

Permalink
[ Print ] Update print result of model summary
Browse files Browse the repository at this point in the history
- 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 <[email protected]>
  • Loading branch information
EunjuYang authored and jijoongmoon committed Oct 4, 2024
1 parent 2807f69 commit f222ecf
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions nntrainer/models/neuralnet.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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 ? '=' : '-')
Expand Down

0 comments on commit f222ecf

Please sign in to comment.