OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Changed how we print the network hash.

This commit is contained in:
Davis King 2017-08-06 08:54:10 -04:00
parent 7b26b2d1ff
commit 0f2753b754
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dlib/dnn/trainer.h
View File

@ -1281,7 +1281,9 @@ namespace dlib
temp.clean();
serialize(temp, sout);
out << " net size: " << sout.str().size()/1024.0/1024.0 << "MB" << endl;
out << " net architecture hash: " << md5(cast_to_string(trainer.get_net())) << endl;
// Don't include the loss params in the hash since we print them on the next line.
// They also aren't really part of the "architecture" of the network.
out << " net architecture hash: " << md5(cast_to_string(trainer.get_net().subnet())) << endl;
out << " loss: " << trainer.get_net().loss_details() << endl;
out << " synchronization file: " << trainer.get_synchronization_file() << endl;