Skip to content

Commit b846f97

Browse files
prabhat00155facebook-github-bot
authored andcommitted
[fbsync] Revert exposing convNormActivation in ops (#5472)
Reviewed By: jdsgomes Differential Revision: D34475311 fbshipit-source-id: 13ea879714fd265166e42409956a73918ac0bd91
1 parent cef9f60 commit b846f97

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)