Minor fix

neural_filters/neural_filter.py

@@ -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

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