diff options
author | Jeff Donahue <jeff.donahue@gmail.com> | 2014-03-13 17:54:10 -0700 |
---|---|---|
committer | Jeff Donahue <jeff.donahue@gmail.com> | 2014-03-19 12:37:31 -0700 |
commit | aee5f54661f3f4678db101cef76aa5ca9b2930bc (patch) | |
tree | b8c73852ecf42ea14aa0af3f83a5a84e15c73a8f /tools/net_speed_benchmark.cpp | |
parent | 748aaff59c6169cc0cb443380b68e4153a47db00 (diff) | |
download | caffeonacl-aee5f54661f3f4678db101cef76aa5ca9b2930bc.tar.gz caffeonacl-aee5f54661f3f4678db101cef76aa5ca9b2930bc.tar.bz2 caffeonacl-aee5f54661f3f4678db101cef76aa5ca9b2930bc.zip |
fix net_speed_benchmark so 'make all' works
Diffstat (limited to 'tools/net_speed_benchmark.cpp')
-rw-r--r-- | tools/net_speed_benchmark.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tools/net_speed_benchmark.cpp b/tools/net_speed_benchmark.cpp index 96d40a2e..43f7b493 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<Blob<float>*>()); + float initial_loss; + caffe_net.Forward(vector<Blob<float>*>(), &initial_loss); + LOG(ERROR) << "Initial loss: " << initial_loss; LOG(ERROR) << "Performing Backward"; - LOG(ERROR) << "Initial loss: " << caffe_net.Backward(); + caffe_net.Backward(); const vector<shared_ptr<Layer<float> > >& layers = caffe_net.layers(); vector<vector<Blob<float>*> >& bottom_vecs = caffe_net.bottom_vecs(); |