Unverified Commit ad0553c0 authored by Vukašin Manojlović's avatar Vukašin Manojlović Committed by GitHub
Browse files

Fix type annotation for num_features in FrozenBatchNorm2d (#2462)

parent 750e38f5
......@@ -50,9 +50,9 @@ class FrozenBatchNorm2d(torch.nn.Module):
def __init__(
self,
num_features: Tuple[int, ...],
num_features: int,
eps: float = 0.,
n: Optional[Tuple[int, ...]] = None,
n: Optional[int] = None,
):
# n=None for backward-compatibility
if n is not None:
......
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