diff options
Diffstat (limited to 'tools/caffe.cpp')
| -rw-r--r-- | tools/caffe.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tools/caffe.cpp b/tools/caffe.cpp index 70b15f89..0b7523fc 100644 --- a/tools/caffe.cpp +++ b/tools/caffe.cpp @@ -187,8 +187,8 @@ int test() { for (int i = 0; i < test_score.size(); ++i) { const std::string& output_name = caffe_net.blob_names()[ caffe_net.output_blob_indices()[test_score_output_id[i]]]; - const float loss_weight = - caffe_net.blob_loss_weights()[caffe_net.output_blob_indices()[i]]; + const float loss_weight = caffe_net.blob_loss_weights()[ + caffe_net.output_blob_indices()[test_score_output_id[i]]]; std::ostringstream loss_msg_stream; const float mean_score = test_score[i] / FLAGS_iterations; if (loss_weight) { |