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

Update dot visitor

Browse Source
This commit is contained in:
Adrià Arrufat 2022-04-07 09:37:41 +09:00 committed by Davis E. King
parent 18fe54b64a
commit e944ea4fcb
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
dlib/dnn/visitors.h
View File

@ -805,9 +805,10 @@ namespace dlib
}
template <typename U, typename E>
void operator()(size_t i, const add_layer<leaky_relu_, U, E>&)
void operator()(size_t i, const add_layer<leaky_relu_, U, E>& l)
{
start_node(i, "leaky_relu");
out << " | { alpha |{" << l.layer_details().get_alpha() << "}}";
end_node();
update(i);
}
@ -860,6 +861,15 @@ namespace dlib
update(i);
}
template <typename U, typename E>
void operator()(size_t i, const add_layer<smelu_, U, E>& l)
{
start_node(i, "smelu");
out << " | { beta |{" << l.layer_details().get_beta() << "}}";
end_node();
update(i);
}
template <typename U, typename E>
void operator()(size_t i, const add_layer<softmax_, U, E>&)
{