Skip to content

Commit 9648e07

Browse files
authored
Revert exposing convNormActivation in ops (#5472) (#5474)
1 parent 211a442 commit 9648e07

File tree

3 files changed

+2
-4
lines changed

3 files changed

+2
-4
lines changed

docs/source/ops.rst

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,5 +45,4 @@ Operators
4545
FeaturePyramidNetwork
4646
StochasticDepth
4747
FrozenBatchNorm2d
48-
ConvNormActivation
4948
SqueezeExcitation

torchvision/models/optical_flow/raft.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from torch import Tensor
77
from torch.nn.modules.batchnorm import BatchNorm2d
88
from torch.nn.modules.instancenorm import InstanceNorm2d
9-
from torchvision.ops import ConvNormActivation
9+
from torchvision.ops.misc import ConvNormActivation
1010

1111
from ..._internally_replaced_utils import load_state_dict_from_url
1212
from ...utils import _log_api_usage_once

torchvision/ops/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from .feature_pyramid_network import FeaturePyramidNetwork
1515
from .focal_loss import sigmoid_focal_loss
1616
from .giou_loss import generalized_box_iou_loss
17-
from .misc import FrozenBatchNorm2d, ConvNormActivation, SqueezeExcitation
17+
from .misc import FrozenBatchNorm2d, SqueezeExcitation
1818
from .poolers import MultiScaleRoIAlign
1919
from .ps_roi_align import ps_roi_align, PSRoIAlign
2020
from .ps_roi_pool import ps_roi_pool, PSRoIPool
@@ -51,7 +51,6 @@
5151
"stochastic_depth",
5252
"StochasticDepth",
5353
"FrozenBatchNorm2d",
54-
"ConvNormActivation",
5554
"SqueezeExcitation",
5655
"generalized_box_iou_loss",
5756
]

0 commit comments

Comments
 (0)