Commit f06b265b authored by Fm's avatar Fm

Added getter for trainer::train_one_step_calls

parent 91163863
...@@ -466,6 +466,12 @@ namespace dlib ...@@ -466,6 +466,12 @@ namespace dlib
return learning_rate_shrink; return learning_rate_shrink;
} }
unsigned long long get_train_one_step_calls (
) const
{
return train_one_step_calls;
}
private: private:
void record_loss(double loss) void record_loss(double loss)
......
...@@ -295,6 +295,17 @@ namespace dlib ...@@ -295,6 +295,17 @@ namespace dlib
get_learning_rate_shrink_factor() to 1. get_learning_rate_shrink_factor() to 1.
!*/ !*/
unsigned long long get_train_one_step_calls (
) const;
/*!
requires
- training process should be done with #train_one_step()
ensures
- Each #train_one_step() call increases this counter. It can be used to understand the training
stage to make some additional processing like snapshotting or extra testing
- This value is serialized/deserialized via synchronization file
!*/
void be_verbose ( void be_verbose (
); );
/*! /*!
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment