Skip to content

GitLab

Commit e944ea4f authored by Adrià Arrufat's avatar Adrià Arrufat Committed by Davis E. King
Browse files

Update dot visitor

parent 18fe54b6
Show 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