diff --git a/elmoformanylangs/modules/highway.py b/elmoformanylangs/modules/highway.py index 934e51c..894945f 100644 --- a/elmoformanylangs/modules/highway.py +++ b/elmoformanylangs/modules/highway.py @@ -43,7 +43,7 @@ def __init__(self, # of the bias vector in each Linear layer. layer.bias[input_dim:].data.fill_(1) - @overrides + @overrides(check_signature=False) def forward(self, inputs: torch.Tensor) -> torch.Tensor: # pylint: disable=arguments-differ current_input = inputs for layer in self._layers: