diff --git a/torch/nn/modules/conv.py b/torch/nn/modules/conv.py index 7280eab37caa..20a1d49619b0 100644 --- a/torch/nn/modules/conv.py +++ b/torch/nn/modules/conv.py @@ -69,7 +69,7 @@ def __init__(self, transposed: bool, output_padding: _size_1_t, groups: int, - bias: Optional[Tensor], + bias: bool, padding_mode: str) -> None: super(_ConvNd, self).__init__() if in_channels % groups != 0: