Several changes:
[jacinto-ai/caffe-jacinto.git] / examples / train_net.cpp
index d84619b2d44c10028fa5fdd215a020c03b677b9a..ce62616b1186aa398c07cb8edfbad6df8ac0405a 100644 (file)
@@ -20,9 +20,6 @@ int main(int argc, char** argv) {
     return 0;
   }
 
-  Caffe::SetDevice(0);
-  Caffe::set_mode(Caffe::GPU);
-
   SolverParameter solver_param;
   ReadProtoFromTextFile(argv[1], &solver_param);