Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dlib
Commits
79a620b1
Commit
79a620b1
authored
Oct 10, 2017
by
Davis King
Browse files
Added assert to improve error messages.
parent
8365bf68
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
0 deletions
+2
-0
dlib/dnn/layers.h
dlib/dnn/layers.h
+2
-0
No files found.
dlib/dnn/layers.h
View file @
79a620b1
...
@@ -1464,6 +1464,8 @@ namespace dlib
...
@@ -1464,6 +1464,8 @@ namespace dlib
template
<
typename
SUBNET
>
template
<
typename
SUBNET
>
void
forward
(
const
SUBNET
&
sub
,
resizable_tensor
&
output
)
void
forward
(
const
SUBNET
&
sub
,
resizable_tensor
&
output
)
{
{
DLIB_CASSERT
(
num_inputs
==
sub
.
get_output
().
nr
()
*
sub
.
get_output
().
nc
()
*
sub
.
get_output
().
k
(),
"The size of the input tensor to this fc layer doesn't match the size the fc layer was trained with."
);
output
.
set_size
(
sub
.
get_output
().
num_samples
(),
num_outputs
);
output
.
set_size
(
sub
.
get_output
().
num_samples
(),
num_outputs
);
auto
w
=
weights
(
params
,
0
);
auto
w
=
weights
(
params
,
0
);
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment