diff --git a/dlib/dnn/trainer.h b/dlib/dnn/trainer.h index 934b74a54..20b51d4cb 100644 --- a/dlib/dnn/trainer.h +++ b/dlib/dnn/trainer.h @@ -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) diff --git a/dlib/dnn/trainer_abstract.h b/dlib/dnn/trainer_abstract.h index 4d0448623..ef6fd9163 100644 --- a/dlib/dnn/trainer_abstract.h +++ b/dlib/dnn/trainer_abstract.h @@ -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 ( ); /*!