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

Fix deserialize logic in loss_multibinary_log (#2547)

Browse Source
This commit is contained in:
Adrià Arrufat 2022-03-23 20:35:56 +09:00 committed by GitHub
parent 1b7053fe9d
commit 3b518f40ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
dlib/dnn/loss.h
View File

@ -877,10 +877,19 @@ namespace dlib
{
std::string version;
deserialize(version, in);
if (version != "loss_multibinary_log_" || version != "loss_multibinary_log_2")
throw serialization_error("Unexpected version found while deserializing dlib::loss_multibinary_log_.");
if (version == "loss_multibinary_log_2")
if (version == "loss_multibinary_log_")
{
item.gamma = 0;
return;
}
else if (version == "loss_multibinary_log_2")
{
deserialize(item.gamma, in);
}
else
{
throw serialization_error("Unexpected version found while deserializing dlib::loss_multibinary_log_.");
}
}
friend std::ostream& operator<<(std::ostream& out, const loss_multibinary_log_& item)