Skip to content

GitLab

Unverified Commit 820fd353 authored by Adrià Arrufat's avatar Adrià Arrufat Committed by GitHub
Browse files

Make dnn_trainer print the minibatch size to ostream (#2236)

parent 375f1172
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
dlib/dnn/trainer.h
View file @ 820fd353
...@@ -1372,6 +1372,7 @@ namespace dlib ...@@ -1372,6 +1372,7 @@ namespace dlib
out << " get_train_one_step_calls(): " << trainer.get_train_one_step_calls() << endl; out << " get_train_one_step_calls(): " << trainer.get_train_one_step_calls() << endl;
out << " synchronization file: " << trainer.get_synchronization_file() << endl; out << " synchronization file: " << trainer.get_synchronization_file() << endl;
out << " trainer.get_solvers()[0]: " << trainer.get_solvers()[0] << endl; out << " trainer.get_solvers()[0]: " << trainer.get_solvers()[0] << endl;
out << " mini batch size: " << trainer.get_mini_batch_size() << endl;
auto sched = trainer.get_learning_rate_schedule(); auto sched = trainer.get_learning_rate_schedule();
if (sched.size() != 0) if (sched.size() != 0)
{ {
......
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