diff --git a/examples/train_net.cpp b/examples/train_net.cpp
index d84619b2d44c10028fa5fdd215a020c03b677b9a..ce62616b1186aa398c07cb8edfbad6df8ac0405a 100644 (file)
--- a/examples/train_net.cpp
+++ b/examples/train_net.cpp
return 0;
}
- Caffe::SetDevice(0);
- Caffe::set_mode(Caffe::GPU);
-
SolverParameter solver_param;
ReadProtoFromTextFile(argv[1], &solver_param);