diff options
author | Jonathan L Long <jonlong@cs.berkeley.edu> | 2014-03-27 15:11:17 -0700 |
---|---|---|
committer | Jonathan L Long <jonlong@cs.berkeley.edu> | 2014-03-27 15:17:51 -0700 |
commit | 71136b0bb770d51de5a0de16d83ed14905593693 (patch) | |
tree | 73acbc5b028d407283c31a04f713ad3293b3a6e6 /tools/net_speed_benchmark.cpp | |
parent | 18d1e96edc4d36781974f5ef522eb0932997bf94 (diff) | |
download | caffeonacl-71136b0bb770d51de5a0de16d83ed14905593693.tar.gz caffeonacl-71136b0bb770d51de5a0de16d83ed14905593693.tar.bz2 caffeonacl-71136b0bb770d51de5a0de16d83ed14905593693.zip |
passing too many args to tool binaries is an error
Diffstat (limited to 'tools/net_speed_benchmark.cpp')
-rw-r--r-- | tools/net_speed_benchmark.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tools/net_speed_benchmark.cpp b/tools/net_speed_benchmark.cpp index db95cc17..bb19b41b 100644 --- a/tools/net_speed_benchmark.cpp +++ b/tools/net_speed_benchmark.cpp @@ -22,7 +22,7 @@ using namespace caffe; // NOLINT(build/namespaces) int main(int argc, char** argv) { int total_iter = 50; - if (argc < 2) { + if (argc < 2 || argc > 5) { LOG(ERROR) << "net_speed_benchmark net_proto [iterations=50]" " [CPU/GPU] [Device_id=0]"; return 0; |