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 4220c19 commit a2c15e1Copy full SHA for a2c15e1
1 file changed
include/graph/graph.hpp
@@ -249,7 +249,7 @@ class Graph {
249
tensors_.push_back(outten_[0]);
250
#endif
251
#ifdef ENABLE_STATISTIC_WEIGHTS
252
- weights_.push_back(layers_[i]->get_weights());
+ weights_.push_back(layers_[current_layer]->get_weights());
253
254
255
inten_ = outten_;
@@ -295,7 +295,7 @@ class Graph {
295
auto elapsed =
296
std::chrono::duration_cast<std::chrono::milliseconds>(end - start);
297
time_.push_back(static_cast<int>(elapsed.count()));
298
- time_layer_.push_back(layers_[i]->getName());
+ time_layer_.push_back(layers_[current_layer]->getName());
299
300
}
301
0 commit comments