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): ...@@ -79,7 +79,7 @@ class NeuralFilter(torch.nn.Module):
if self.hidden_size != input_var.size(-1): if self.hidden_size != input_var.size(-1):
raise RuntimeError( raise RuntimeError(
'input.size(-1) must be equal to hidden_size. Expected {}, got {}'.format( '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: if is_input_packed:
mini_batch = int(batch_sizes[0]) mini_batch = int(batch_sizes[0])
......
neural_filters/neural_filter_2CC.py
View file @ b00df262
...@@ -94,7 +94,7 @@ class NeuralFilter2CC(torch.nn.Module): ...@@ -94,7 +94,7 @@ class NeuralFilter2CC(torch.nn.Module):
if self.hidden_size != input_var.size(-1): if self.hidden_size != input_var.size(-1):
raise RuntimeError( raise RuntimeError(
'input.size(-1) must be equal to hidden_size. Expected {}, got {}'.format( '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: if is_input_packed:
mini_batch = int(batch_sizes[0]) 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