Skip to content

Commit ad0553c

Browse files
authored
Fix type annotation for num_features in FrozenBatchNorm2d (#2462)
1 parent 750e38f commit ad0553c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchvision/ops/misc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,9 @@ class FrozenBatchNorm2d(torch.nn.Module):
5050

5151
def __init__(
5252
self,
53-
num_features: Tuple[int, ...],
53+
num_features: int,
5454
eps: float = 0.,
55-
n: Optional[Tuple[int, ...]] = None,
55+
n: Optional[int] = None,
5656
):
5757
# n=None for backward-compatibility
5858
if n is not None:

0 commit comments

Comments
 (0)