Several changes:
[jacinto-ai/caffe-jacinto.git] / examples / finetune_net.cpp
index d35b425c6196234f377564dca0d57f9eea34acbd..559715d6707c9fcdef3dea616fad07832a2470f9 100644 (file)
@@ -19,9 +19,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);