Commit 4a2f0192 authored by Davis King's avatar Davis King

Fixed a bug in the repeat layer that caused it to throw exceptions

in some cases.
parent 85d421c3
...@@ -81,7 +81,6 @@ namespace dlib ...@@ -81,7 +81,6 @@ namespace dlib
resizable_tensor& resizable_tensor&
) const ) const
{ {
DLIB_CASSERT(false,"This function should never be called");
} }
friend void serialize(const repeat_input_layer&, std::ostream&){} friend void serialize(const repeat_input_layer&, std::ostream&){}
...@@ -1681,6 +1680,11 @@ namespace dlib ...@@ -1681,6 +1680,11 @@ namespace dlib
) const ) const
{ {
subnetwork.to_tensor(ibegin,iend,data); subnetwork.to_tensor(ibegin,iend,data);
// call to_tensor on the networks in details just to populate the
// _sample_expansion_factor values in those networks. Other than that this
// call is a noop.
for (auto& d : details)
d.to_tensor(ibegin, iend, data);
} }
template <typename forward_iterator> template <typename forward_iterator>
......
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