Skip to content

N
neural_filters
Commit b00df262 authored by M. François's avatar M. François
Browse files
Options

Minor fix

parent c4ad480f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
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])
......
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