We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2a2b9df commit e717f5eCopy full SHA for e717f5e
1 file changed
include/graph/graph.hpp
@@ -224,7 +224,7 @@ class Graph {
224
tensors_.push_back(outten_[0]);
225
#endif
226
#ifdef ENABLE_STATISTIC_WEIGHTS
227
- weights_.push_back(layers_[i]->get_weights());
+ weights_.push_back(layers_[current_layer]->get_weights());
228
229
230
inten_ = outten_;
@@ -270,7 +270,7 @@ class Graph {
270
auto elapsed =
271
std::chrono::duration_cast<std::chrono::milliseconds>(end - start);
272
time_.push_back(static_cast<int>(elapsed.count()));
273
- time_layer_.push_back(layers_[i]->getName());
+ time_layer_.push_back(layers_[current_layer]->getName());
274
275
}
276
0 commit comments