Handle tag layers as inputs to disable_duplicative_biases (#2416)

e0a5725f · Adrià Arrufat · GitHub · 5bd837d1 · e0a5725f
Unverified Commit e0a5725f authored Aug 19, 2021 by Adrià Arrufat Committed by GitHub Aug 19, 2021
Show whitespace changes
Inline Side-by-side

Showing with 15 additions and 4 deletions

dlib/dnn/layers.h dlib/dnn/layers.h +15 -4

No files found.
--- a/dlib/dnn/layers.h
+++ b/dlib/dnn/layers.h
@@ -1757,13 +1757,24 @@ namespace dlib
            }
            // handle input repeat layer with tag case
-            template <layer_mode mode, unsigned long ID, typename E, typename F>
+            template <layer_mode mode, unsigned long ID, typename E>
-            void disable_input_bias(add_layer<bn_<mode>, add_tag_layer<ID, impl::repeat_input_layer, E>, F>& )
+            void disable_input_bias(add_layer<bn_<mode>, add_tag_layer<ID, impl::repeat_input_layer>, E>& )
            {
            }
-            template <unsigned long ID, typename E, typename F>
+            template <unsigned long ID, typename E>
-            void disable_input_bias(add_layer<layer_norm_, add_tag_layer<ID, impl::repeat_input_layer, E>, F>& )
+            void disable_input_bias(add_layer<layer_norm_, add_tag_layer<ID, impl::repeat_input_layer>, E>& )
+            {
+            }
+            // handle tag layer case
+            template <layer_mode mode, unsigned long ID, typename U, typename E>
+            void disable_input_bias(add_layer<bn_<mode>, add_tag_layer<ID, U>, E>& )
+            {
+            }
+            template <unsigned long ID, typename U, typename E>
+            void disable_input_bias(add_layer<layer_norm_, add_tag_layer<ID, U>, E>& )
            {
            }