fixed solver constructor in train_net.cpp

lhavipl · Sep 1, 2014 · 8b3dde0 · 8b3dde0
1 parent ed8b1da
commit 8b3dde0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tools/train_net.cpp b/tools/train_net.cpp
@@ -22,8 +22,7 @@ int main(int argc, char** argv) {
   ReadProtoFromTextFileOrDie(argv[1], &solver_param);
 
   LOG(INFO) << "Starting Optimization";
-  shared_ptr<Solver<float> > solver =
-      (shared_ptr<Solver<float> >) GetSolver<float>(solver_param);
+  shared_ptr<Solver<float> > solver(GetSolver<float>(solver_param));
   if (argc == 3) {
     LOG(INFO) << "Resuming from " << argv[2];
     solver->Solve(argv[2]);