[ Web Proxy ]
URL:
Viewing: https://raw.githubusercontent.com/rootkit/CaffeOnACL/master/src/caffe/solver.cpp [Back]  [Original]

#include 

#include 
#include 

#include "caffe/solver.hpp"
#include "caffe/util/format.hpp"
#include "caffe/util/hdf5.hpp"
#include "caffe/util/io.hpp"
#include "caffe/util/upgrade_proto.hpp"

namespace caffe {

template
void Solver::SetActionFunction(ActionCallback func) {
  action_request_function_ = func;
}

template
SolverAction::Enum Solver::GetRequestedAction() {
  if (action_request_function_) {
    // If the external request function has been set, call it.
    return action_request_function_();
  }
  return SolverAction::NONE;
}

template 
Solver::Solver(const SolverParameter& param)
    : net_(), callbacks_(), requested_early_exit_(false) {
  Init(param);
}

template 
Solver::Solver(const string& param_file)
    : net_(), callbacks_(), requested_early_exit_(false) {
  SolverParameter param;
  ReadSolverParamsFromTextFileOrDie(param_file, ¶m);
  Init(param);
}

template 
void Solver::Init(const SolverParameter& param) {
  LOG_IF(INFO, Caffe::root_solver()) 

Web Proxy Viewer  |  New URL  |  Original Page