stochastic pooling test

[jacinto-ai/caffe-jacinto.git] / examples / demo_mnist.cpp

diff --git a/examples/demo_mnist.cpp b/examples/demo_mnist.cpp
index 9eb6204ca0de53a6ffe7f61188aa9a77fa219988..11d3fc518b999e270f6b2a19252da45a72000a75 100644 (file)
--- a/examples/demo_mnist.cpp
+++ b/examples/demo_mnist.cpp
@@ -25,6 +25,7 @@ int main(int argc, char** argv) {
     return 0;
   }
   google::InitGoogleLogging(argv[0]);
+  Caffe::DeviceQuery();
 
   if (argc == 4 && strcmp(argv[3], "GPU") == 0) {
     LOG(ERROR) << "Using GPU";