From 1875eb63b760fa3367a20bb9b3fabd3ab6cc1ca2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Vuka=C5=A1in=20Manojlovi=C4=87?= <iamvukasin@gmail.com>
Date: Sat, 11 Jul 2020 18:00:31 +0200
Subject: [PATCH] Fix type annotation for num_features in FrozenBatchNorm2d

---
 torchvision/ops/misc.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/torchvision/ops/misc.py b/torchvision/ops/misc.py
index a0f15653145..17e69c506d8 100644
--- 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: