Skip to content

Commit 6fa0843

Browse files
committed
fix cout
1 parent 9f73bd9 commit 6fa0843

File tree

1 file changed

+1
-17
lines changed

1 file changed

+1
-17
lines changed

include/graph/graph.hpp

Lines changed: 1 addition & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -210,27 +210,11 @@ class Graph {
210210
new_branch.distribution = dis;
211211
}
212212
if (layers_[current_layer]->getName() == kSplit) {
213-
std::cout << "=== SPLIT LAYER DEBUG INFO ===" << std::endl;
214-
std::cout << "Split layer #" << current_layer
215-
<< " outputs: " << outten_.size() << std::endl;
216-
217213
for (size_t out_idx = 0; out_idx < outten_.size(); ++out_idx) {
218-
std::cout << " Output " << out_idx << ": shape [";
219214
for (size_t d = 0; d < outten_[out_idx].get_shape().dims(); ++d) {
220-
std::cout << outten_[out_idx].get_shape()[d];
221-
if (d < outten_[out_idx].get_shape().dims() - 1) std::cout << ", ";
222-
}
223-
std::cout << "]" << std::endl;
224-
225-
std::cout << " Distribution for this output: ";
226-
for (const auto& dist : new_branch.distribution) {
227-
if (dist.second == static_cast<int>(out_idx)) {
228-
std::cout << "-> Layer #" << dist.first << " ";
229-
}
215+
if (d < outten_[out_idx].get_shape().dims() - 1) std::cout << "";
230216
}
231-
std::cout << std::endl;
232217
}
233-
std::cout << "=============================" << std::endl;
234218
}
235219
branch_list_.push_back(new_branch);
236220

0 commit comments

Comments
 (0)