mirror of https://github.com/davisking/dlib.git
Merge pull request #125 from e-fominov/dnn_trainer_get_step
Added getter for trainer::train_one_step_calls
This commit is contained in:
commit
7a31806baa
|
@ -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)
|
||||
|
|
|
@ -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 (
|
||||
);
|
||||
/*!
|
||||
|
|
Loading…
Reference in New Issue