Commit b00df262 authored by M. François's avatar M. François

Minor fix

parent c4ad480f
...@@ -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])
......
...@@ -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