From aee5f54661f3f4678db101cef76aa5ca9b2930bc Mon Sep 17 00:00:00 2001 From: Jeff Donahue Date: Thu, 13 Mar 2014 17:54:10 -0700 Subject: [PATCH] fix net_speed_benchmark so 'make all' works --- tools/net_speed_benchmark.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tools/net_speed_benchmark.cpp b/tools/net_speed_benchmark.cpp index 96d40a2..43f7b49 100644 --- a/tools/net_speed_benchmark.cpp +++ b/tools/net_speed_benchmark.cpp @@ -58,9 +58,11 @@ int main(int argc, char** argv) { LOG(ERROR) << "Performing Forward"; // Note that for the speed benchmark, we will assume that the network does // not take any input blobs. - caffe_net.Forward(vector*>()); + float initial_loss; + caffe_net.Forward(vector*>(), &initial_loss); + LOG(ERROR) << "Initial loss: " << initial_loss; LOG(ERROR) << "Performing Backward"; - LOG(ERROR) << "Initial loss: " << caffe_net.Backward(); + caffe_net.Backward(); const vector > >& layers = caffe_net.layers(); vector*> >& bottom_vecs = caffe_net.bottom_vecs(); -- 2.7.4