"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "c91272d63155faaf35e62f3e2e8a3b17fa28610a"
Commit 0bb6ce36 authored by Davis King's avatar Davis King
Browse files

dnn_trainer prints number of steps executed when print to ostream

parent 76cc8e3b
...@@ -1372,6 +1372,7 @@ namespace dlib ...@@ -1372,6 +1372,7 @@ namespace dlib
out << " net architecture hash: " << md5(cast_to_string(trainer.get_net().subnet())) << endl; out << " net architecture hash: " << md5(cast_to_string(trainer.get_net().subnet())) << endl;
out << " loss: " << trainer.get_net().loss_details() << endl; out << " loss: " << trainer.get_net().loss_details() << 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;
auto sched = trainer.get_learning_rate_schedule(); auto sched = trainer.get_learning_rate_schedule();
......
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