Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include "pch.h"
- #define CPU_ONLY
- #include <iostream>
- #include <caffe/caffe.hpp>
- #include <memory>
- #include "caffe/layers/memory_data_layer.hpp"
- #include "caffe/solver_factory.hpp"
- using namespace caffe;
- int main()
- {
- Caffe::set_mode(Caffe::CPU);
- // Create the Solver object.
- SolverParameter solverParam;
- ReadSolverParamsFromTextFileOrDie("solver.prototxt", &solverParam);
- boost::shared_ptr<Solver<float>> solver(SolverRegistry<float>::CreateSolver(solverParam));
- ...
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement