Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/ops/misc.py
View file @ ad0553c0
...@@ -50,9 +50,9 @@ class FrozenBatchNorm2d(torch.nn.Module): ...@@ -50,9 +50,9 @@ class FrozenBatchNorm2d(torch.nn.Module):
def __init__( def __init__(
self, self,
num_features: Tuple[int, ...], num_features: int,
eps: float = 0., eps: float = 0.,
n: Optional[Tuple[int, ...]] = None, n: Optional[int] = None,
): ):
# n=None for backward-compatibility # n=None for backward-compatibility
if n is not None: 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