Unverified Commit 973de8ac authored by Adrià Arrufat's avatar Adrià Arrufat Committed by GitHub
Browse files

Fix disable_duplicative_biases when the input is a skip layer (#2367)

* Fix disable_duplicative_biases when the input is a skip layer

* fix template parameters
parent 4a51017c
......@@ -1767,6 +1767,17 @@ namespace dlib
{
}
// handle skip layer case
template <layer_mode mode, template <typename> class TAG, typename U, typename E>
void disable_input_bias(add_layer<bn_<mode>, add_skip_layer<TAG, U>, E>& )
{
}
template <template <typename> class TAG, typename U, typename E>
void disable_input_bias(add_layer<layer_norm_, add_skip_layer<TAG, U>, E>& )
{
}
template<typename input_layer_type>
void operator()(size_t , input_layer_type& ) const
{
......
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