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
b084076e
Commit
b084076e
authored
Nov 20, 2015
by
Davis King
Browse files
Added batch normalization layer skeleton
parent
627db9e1
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
37 additions
and
0 deletions
+37
-0
dlib/dnn/layers.h
dlib/dnn/layers.h
+37
-0
No files found.
dlib/dnn/layers.h
View file @
b084076e
...
@@ -53,6 +53,43 @@ namespace dlib
...
@@ -53,6 +53,43 @@ namespace dlib
template
<
typename
SUBNET
>
template
<
typename
SUBNET
>
using
con
=
add_layer
<
con_
,
SUBNET
>
;
using
con
=
add_layer
<
con_
,
SUBNET
>
;
// ----------------------------------------------------------------------------------------
class
bn_
{
public:
bn_
()
{}
template
<
typename
SUBNET
>
void
setup
(
const
SUBNET
&
sub
)
{
// TODO
}
template
<
typename
SUBNET
>
void
forward
(
const
SUBNET
&
sub
,
resizable_tensor
&
output
)
{
// TODO
}
template
<
typename
SUBNET
>
void
backward
(
const
tensor
&
gradient_input
,
SUBNET
&
sub
,
tensor
&
params_grad
)
{
// TODO
}
const
tensor
&
get_layer_params
()
const
{
return
params
;
}
tensor
&
get_layer_params
()
{
return
params
;
}
private:
resizable_tensor
params
;
};
template
<
typename
SUBNET
>
using
bn
=
add_layer
<
bn_
,
SUBNET
>
;
// ----------------------------------------------------------------------------------------
// ----------------------------------------------------------------------------------------
class
fc_
class
fc_
...
...
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