Minor fix in the network size format (#2303)

Since we are dividing by 1024, the unit should be MiB instead of MB.
I also added a space between the number and the unit
This commit is contained in:
Adrià Arrufat 2021-02-09 22:07:24 +09:00 committed by GitHub
parent 9f6aefc0db
commit 42e6ace845
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1363,7 +1363,7 @@ namespace dlib
net_type temp = trainer.get_net(); // make a copy so that we can clean it without mutating the trainer's net.
temp.clean();
serialize(temp, sout);
out << " net size: " << sout.str().size()/1024.0/1024.0 << "MB" << endl;
out << " net size: " << sout.str().size()/1024.0/1024.0 << " MiB" << 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;