File tree Expand file tree Collapse file tree 1 file changed +1
-17
lines changed Expand file tree Collapse file tree 1 file changed +1
-17
lines changed Original file line number Diff line number Diff line change @@ -210,27 +210,11 @@ class Graph {
210
210
new_branch.distribution = dis;
211
211
}
212
212
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
-
217
213
for (size_t out_idx = 0 ; out_idx < outten_.size (); ++out_idx) {
218
- std::cout << " Output " << out_idx << " : shape [" ;
219
214
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 << " " ;
230
216
}
231
- std::cout << std::endl;
232
217
}
233
- std::cout << " =============================" << std::endl;
234
218
}
235
219
branch_list_.push_back (new_branch);
236
220
You can’t perform that action at this time.
0 commit comments