Merge pull request #125 from e-fominov/dnn_trainer_get_step

Added getter for trainer::train_one_step_calls
This commit is contained in:
Davis E. King 2016-05-30 07:31:17 -04:00
commit 7a31806baa
2 changed files with 17 additions and 0 deletions

View File

@ -466,6 +466,12 @@ namespace dlib
return learning_rate_shrink;
}
unsigned long long get_train_one_step_calls (
) const
{
return train_one_step_calls;
}
private:
void record_loss(double loss)

View File

@ -295,6 +295,17 @@ namespace dlib
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 (
);
/*!