From 1f3bb9d35975e7493fa914344b87ba743d168bcd Mon Sep 17 00:00:00 2001 From: Francisco Massa Date: Wed, 20 May 2020 14:41:03 +0200 Subject: [PATCH] Deprecate Conv2d, ConvTranspose2d and BatchNorm2d (#2244) * Deprecate Conv2d, ConvTranspose2d and BatchNorm * Fix lint --- torchvision/ops/misc.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/torchvision/ops/misc.py b/torchvision/ops/misc.py index 3003796303f..57e53fbd85e 100644 --- a/torchvision/ops/misc.py +++ b/torchvision/ops/misc.py @@ -71,6 +71,30 @@ def forward(self, x): return _new_empty_tensor(x, output_shape) +class Conv2d(torch.nn.Conv2d): + def __init__(self, *args, **kwargs): + super().__init__(*args, **kwargs) + warnings.warn( + "torchvision.ops.misc.Conv2d is deprecated and will be " + "removed in future versions, use torch.nn.Conv2d instead.", FutureWarning) + + +class ConvTranspose2d(torch.nn.ConvTranspose2d): + def __init__(self, *args, **kwargs): + super().__init__(*args, **kwargs) + warnings.warn( + "torchvision.ops.misc.ConvTranspose2d is deprecated and will be " + "removed in future versions, use torch.nn.ConvTranspose2d instead.", FutureWarning) + + +class BatchNorm2d(torch.nn.BatchNorm2d): + def __init__(self, *args, **kwargs): + super().__init__(*args, **kwargs) + warnings.warn( + "torchvision.ops.misc.BatchNorm2d is deprecated and will be " + "removed in future versions, use torch.nn.BatchNorm2d instead.", FutureWarning) + + def _check_size_scale_factor(dim, size, scale_factor): # type: (int, Optional[List[int]], Optional[float]) -> None if size is None and scale_factor is None: