From 3345aa409a1105a7d2c488dafd41ed3623a91b0c Mon Sep 17 00:00:00 2001 From: James Thewlis Date: Mon, 17 Mar 2014 20:50:41 +0000 Subject: [PATCH] Fix wrong argv check for whether a backward pass should be done --- tools/dump_network.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/dump_network.cpp b/tools/dump_network.cpp index 386b8a3..6448ce6 100644 --- 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 -- 2.7.4