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