projects
/
platform
/
core
/
ml
/
nntrainer.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
2807f69
)
[ Print ] Update print result of model summary
author
Eunju Yang
<ej.yang@samsung.com>
Wed, 2 Oct 2024 05:31:32 +0000
(14:31 +0900)
committer
Jijoong 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
patch
|
blob
|
history
diff --git
a/nntrainer/models/neuralnet.cpp
b/nntrainer/models/neuralnet.cpp
index d0e542825fa736265938cb0ae042c07fd3415227..fd7dd7f032d5ddef3eb8ed0ba377697c076d6640 100644
(file)
--- 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 ? '=' : '-')