Fix type annotation for num_features in FrozenBatchNorm2d (#2462)

pytorch · Jul 12, 2020 · ad0553c · ad0553c
1 parent 750e38f
commit ad0553c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchvision/ops/misc.py b/torchvision/ops/misc.py
@@ -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: