Fix wrong argv check for whether a backward pass should be done

diff --git a/tools/dump_network.cpp b/tools/dump_network.cpp
index 386b8a3..6448ce6 100644 (file)
--- a/tools/dump_network.cpp
+++ b/tools/dump_network.cpp
@@ -59,7 +59,7 @@ int main(int argc, char** argv) {
   // Run the network without training.
   LOG(ERROR) << "Performing Forward";
   caffe_net->Forward(input_vec);
-  if (argc > 4 && strcmp(argv[4], "1")) {
+  if (argc > 5 && strcmp(argv[5], "1") == 0) {
     LOG(ERROR) << "Performing Backward";
     caffe_net->Backward();
     // Dump the network