Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "b1e69105d4ad4b998f167e84ef928bf60df63292"
Commit e944ea4f authored by Adrià Arrufat's avatar Adrià Arrufat Committed by Davis E. King
Browse files

Update dot visitor

parent 18fe54b6
Hide whitespace changes
Inline Side-by-side
Showing with 11 additions and 1 deletion
dlib/dnn/visitors.h
View file @ e944ea4f
...@@ -805,9 +805,10 @@ namespace dlib ...@@ -805,9 +805,10 @@ namespace dlib
} }
template <typename U, typename E> 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"); start_node(i, "leaky_relu");
out << " | { alpha |{" << l.layer_details().get_alpha() << "}}";
end_node(); end_node();
update(i); update(i);
} }
...@@ -860,6 +861,15 @@ namespace dlib ...@@ -860,6 +861,15 @@ namespace dlib
update(i); 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> template <typename U, typename E>
void operator()(size_t i, const add_layer<softmax_, U, E>&) void operator()(size_t i, const add_layer<softmax_, U, E>&)
{ {
......
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