Update dot visitor

e944ea4f · Adrià Arrufat · Davis E. King · 18fe54b6 · e944ea4f
Commit e944ea4f authored Apr 07, 2022 by Adrià Arrufat Committed by Davis E. King Apr 12, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 11 additions and 1 deletion

dlib/dnn/visitors.h dlib/dnn/visitors.h +11 -1

No files found.
--- a/dlib/dnn/visitors.h
+++ b/dlib/dnn/visitors.h
@@ -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>&)
            {