Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
software
neural_filters
Commits
b00df262
Commit
b00df262
authored
Jun 27, 2018
by
M. François
Browse files
Minor fix
parent
c4ad480f
Changes
2
Hide whitespace changes
Inline
Side-by-side
neural_filters/neural_filter.py
View file @
b00df262
...
...
@@ -79,7 +79,7 @@ class NeuralFilter(torch.nn.Module):
if
self
.
hidden_size
!=
input_var
.
size
(
-
1
):
raise
RuntimeError
(
'input.size(-1) must be equal to hidden_size. Expected {}, got {}'
.
format
(
self
.
input
_size
,
input_var
.
size
(
-
1
)))
self
.
hidden
_size
,
input_var
.
size
(
-
1
)))
if
is_input_packed
:
mini_batch
=
int
(
batch_sizes
[
0
])
...
...
neural_filters/neural_filter_2CC.py
View file @
b00df262
...
...
@@ -94,7 +94,7 @@ class NeuralFilter2CC(torch.nn.Module):
if
self
.
hidden_size
!=
input_var
.
size
(
-
1
):
raise
RuntimeError
(
'input.size(-1) must be equal to hidden_size. Expected {}, got {}'
.
format
(
self
.
input
_size
,
input_var
.
size
(
-
1
)))
self
.
hidden
_size
,
input_var
.
size
(
-
1
)))
if
is_input_packed
:
mini_batch
=
int
(
batch_sizes
[
0
])
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new 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