diff options
author | James Thewlis <jamt9000@gmail.com> | 2014-03-17 20:50:41 +0000 |
---|---|---|
committer | James Thewlis <jamt9000@gmail.com> | 2014-03-17 20:50:41 +0000 |
commit | 3345aa409a1105a7d2c488dafd41ed3623a91b0c (patch) | |
tree | cac93fc8f5ae0f296bcc37e2a7e984d033507384 /tools | |
parent | 67f6910fc77dab65f10b3d627089e0911c24ad65 (diff) | |
download | caffeonacl-3345aa409a1105a7d2c488dafd41ed3623a91b0c.tar.gz caffeonacl-3345aa409a1105a7d2c488dafd41ed3623a91b0c.tar.bz2 caffeonacl-3345aa409a1105a7d2c488dafd41ed3623a91b0c.zip |
Fix wrong argv check for whether a backward pass should be done
Diffstat (limited to 'tools')
-rw-r--r-- | tools/dump_network.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tools/dump_network.cpp b/tools/dump_network.cpp index 386b8a39..6448ce6c 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 |