Commit f42b4ad2 authored by Davis King's avatar Davis King

Fixed bug in add_tag_layer that prevented it from working when placed

immediately after the input layer.
parent 5165ae1b
...@@ -1580,7 +1580,7 @@ namespace dlib ...@@ -1580,7 +1580,7 @@ namespace dlib
public: public:
typedef INPUT_LAYER subnet_type; typedef INPUT_LAYER subnet_type;
typedef typename subnet_type::input_type input_type; typedef typename subnet_type::input_type input_type;
const static size_t num_layers = 0; const static size_t num_layers = 1;
const static unsigned int sample_expansion_factor = subnet_type::sample_expansion_factor; const static unsigned int sample_expansion_factor = subnet_type::sample_expansion_factor;
static_assert(sample_expansion_factor >= 1, static_assert(sample_expansion_factor >= 1,
"The input layer can't produce fewer output tensors than there are inputs."); "The input layer can't produce fewer output tensors than there are inputs.");
......
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