Advertisement
Guest User

Untitled

a guest
Jun 15th, 2019
107
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.50 KB | None | 0 0
  1. #include "pch.h"
  2.  
  3. #define CPU_ONLY
  4.  
  5. #include <iostream>
  6.  
  7. #include <caffe/caffe.hpp>
  8. #include <memory>
  9. #include "caffe/layers/memory_data_layer.hpp"
  10. #include "caffe/solver_factory.hpp"
  11.  
  12. using namespace caffe;
  13.  
  14. int main()
  15. {
  16.     Caffe::set_mode(Caffe::CPU);
  17.  
  18.     // Create the Solver object.
  19.  
  20.     SolverParameter solverParam;
  21.     ReadSolverParamsFromTextFileOrDie("solver.prototxt", &solverParam);
  22.     boost::shared_ptr<Solver<float>> solver(SolverRegistry<float>::CreateSolver(solverParam));
  23.  
  24.     ...
  25. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement