NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Adrià Arrufat 2020-11-17 12:17:29 +09:00 committed by GitHub
parent 375f117222
commit 820fd353d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dlib/dnn/trainer.h
View File

@ -1372,6 +1372,7 @@ namespace dlib
out << " get_train_one_step_calls(): " << trainer.get_train_one_step_calls() << endl;
out << " synchronization file: " << trainer.get_synchronization_file() << 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();
if (sched.size() != 0)
{