Caffe
Public Member Functions | Protected Member Functions | List of all members
caffe::RMSPropSolver< Dtype > Class Template Reference
Inheritance diagram for caffe::RMSPropSolver< Dtype >:
caffe::SGDSolver< Dtype > caffe::Solver< Dtype >

Public Member Functions

 RMSPropSolver (const SolverParameter &param)
 
 RMSPropSolver (const string &param_file)
 
virtual const char * type () const
 Returns the solver type.
 
- Public Member Functions inherited from caffe::SGDSolver< Dtype >
 SGDSolver (const SolverParameter &param)
 
 SGDSolver (const string &param_file)
 
const vector< shared_ptr< Blob< Dtype > > > & history ()
 
- Public Member Functions inherited from caffe::Solver< Dtype >
 Solver (const SolverParameter &param)
 
 Solver (const string &param_file)
 
void Init (const SolverParameter &param)
 
void InitTrainNet ()
 
void InitTestNets ()
 
void SetActionFunction (ActionCallback func)
 
SolverAction::Enum GetRequestedAction ()
 
virtual void Solve (const char *resume_file=NULL)
 
void Solve (const string resume_file)
 
void Step (int iters)
 
void Restore (const char *resume_file)
 
void Snapshot ()
 
const SolverParameter & param () const
 
shared_ptr< Net< Dtype > > net ()
 
const vector< shared_ptr< Net< Dtype > > > & test_nets ()
 
int iter () const
 
const vector< Callback * > & callbacks () const
 
void add_callback (Callback *value)
 
void CheckSnapshotWritePermissions ()
 

Protected Member Functions

virtual void ComputeUpdateValue (int param_id, Dtype rate)
 
void constructor_sanity_check ()
 
 DISABLE_COPY_AND_ASSIGN (RMSPropSolver)
 
- Protected Member Functions inherited from caffe::SGDSolver< Dtype >
void PreSolve ()
 
Dtype GetLearningRate ()
 
virtual void ApplyUpdate ()
 
virtual void Normalize (int param_id)
 
virtual void Regularize (int param_id)
 
virtual void ClipGradients ()
 
virtual void SnapshotSolverState (const string &model_filename)
 
virtual void SnapshotSolverStateToBinaryProto (const string &model_filename)
 
virtual void SnapshotSolverStateToHDF5 (const string &model_filename)
 
virtual void RestoreSolverStateFromHDF5 (const string &state_file)
 
virtual void RestoreSolverStateFromBinaryProto (const string &state_file)
 
 DISABLE_COPY_AND_ASSIGN (SGDSolver)
 
- Protected Member Functions inherited from caffe::Solver< Dtype >
string SnapshotFilename (const string extension)
 
string SnapshotToBinaryProto ()
 
string SnapshotToHDF5 ()
 
void TestAll ()
 
void Test (const int test_net_id=0)
 
void DisplayOutputBlobs (const int net_id)
 
void UpdateSmoothedLoss (Dtype loss, int start_iter, int average_loss)
 
 DISABLE_COPY_AND_ASSIGN (Solver)
 

Additional Inherited Members

- Protected Attributes inherited from caffe::SGDSolver< Dtype >
vector< shared_ptr< Blob< Dtype > > > history_
 
vector< shared_ptr< Blob< Dtype > > > update_
 
vector< shared_ptr< Blob< Dtype > > > temp_
 
- Protected Attributes inherited from caffe::Solver< Dtype >
SolverParameter param_
 
int iter_
 
int current_step_
 
shared_ptr< Net< Dtype > > net_
 
vector< shared_ptr< Net< Dtype > > > test_nets_
 
vector< Callback * > callbacks_
 
vector< Dtype > losses_
 
Dtype smoothed_loss_
 
ActionCallback action_request_function_
 
bool requested_early_exit_
 
Timer iteration_timer_
 
float iterations_last_
 

The documentation for this class was generated from the following files: