Skip to content

Commit 90dce1a

Browse files
committed
Temporarily undo typing patch.
1 parent abab567 commit 90dce1a

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

torchvision/models/detection/transform.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,7 @@ class GeneralizedRCNNTransform(nn.Module):
7272
It returns a ImageList for the inputs, and a List[Dict[Tensor]] for the targets
7373
"""
7474

75-
def __init__(self, min_size: int, max_size: int, image_mean: List[float], image_std: List[float],
76-
size_divisible: int = 32, fixed_size: Optional[Tuple[int, int]] = None):
75+
def __init__(self, min_size, max_size, image_mean, image_std, size_divisible=32, fixed_size=None):
7776
super(GeneralizedRCNNTransform, self).__init__()
7877
if not isinstance(min_size, (list, tuple)):
7978
min_size = (min_size,)

0 commit comments

Comments
 (0)