summaryrefslogtreecommitdiff
path: root/tools/test_net.cpp
diff options
context:
space:
mode:
authorJonathan L Long <jonlong@cs.berkeley.edu>2014-03-27 15:11:17 -0700
committerJonathan L Long <jonlong@cs.berkeley.edu>2014-03-27 15:17:51 -0700
commit71136b0bb770d51de5a0de16d83ed14905593693 (patch)
tree73acbc5b028d407283c31a04f713ad3293b3a6e6 /tools/test_net.cpp
parent18d1e96edc4d36781974f5ef522eb0932997bf94 (diff)
downloadcaffeonacl-71136b0bb770d51de5a0de16d83ed14905593693.tar.gz
caffeonacl-71136b0bb770d51de5a0de16d83ed14905593693.tar.bz2
caffeonacl-71136b0bb770d51de5a0de16d83ed14905593693.zip
passing too many args to tool binaries is an error
Diffstat (limited to 'tools/test_net.cpp')
-rw-r--r--tools/test_net.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/tools/test_net.cpp b/tools/test_net.cpp
index 63922314..75d42d66 100644
--- a/tools/test_net.cpp
+++ b/tools/test_net.cpp
@@ -17,7 +17,7 @@
using namespace caffe; // NOLINT(build/namespaces)
int main(int argc, char** argv) {
- if (argc < 4) {
+ if (argc < 4 || argc > 5) {
LOG(ERROR) << "test_net net_proto pretrained_net_proto iterations "
<< "[CPU/GPU]";
return 0;