Unverified Commit 55a8300b authored by Joao Gomes's avatar Joao Gomes Committed by GitHub
Browse files

Revert exposing convNormActivation in ops (#5472)

parent 19d8987d
...@@ -45,5 +45,4 @@ Operators ...@@ -45,5 +45,4 @@ Operators
FeaturePyramidNetwork FeaturePyramidNetwork
StochasticDepth StochasticDepth
FrozenBatchNorm2d FrozenBatchNorm2d
ConvNormActivation
SqueezeExcitation SqueezeExcitation
...@@ -6,7 +6,7 @@ import torch.nn.functional as F ...@@ -6,7 +6,7 @@ import torch.nn.functional as F
from torch import Tensor from torch import Tensor
from torch.nn.modules.batchnorm import BatchNorm2d from torch.nn.modules.batchnorm import BatchNorm2d
from torch.nn.modules.instancenorm import InstanceNorm2d from torch.nn.modules.instancenorm import InstanceNorm2d
from torchvision.ops import ConvNormActivation from torchvision.ops.misc import ConvNormActivation
from ..._internally_replaced_utils import load_state_dict_from_url from ..._internally_replaced_utils import load_state_dict_from_url
from ...utils import _log_api_usage_once from ...utils import _log_api_usage_once
......
...@@ -14,7 +14,7 @@ from .deform_conv import deform_conv2d, DeformConv2d ...@@ -14,7 +14,7 @@ from .deform_conv import deform_conv2d, DeformConv2d
from .feature_pyramid_network import FeaturePyramidNetwork from .feature_pyramid_network import FeaturePyramidNetwork
from .focal_loss import sigmoid_focal_loss from .focal_loss import sigmoid_focal_loss
from .giou_loss import generalized_box_iou_loss from .giou_loss import generalized_box_iou_loss
from .misc import FrozenBatchNorm2d, ConvNormActivation, SqueezeExcitation from .misc import FrozenBatchNorm2d, SqueezeExcitation
from .poolers import MultiScaleRoIAlign from .poolers import MultiScaleRoIAlign
from .ps_roi_align import ps_roi_align, PSRoIAlign from .ps_roi_align import ps_roi_align, PSRoIAlign
from .ps_roi_pool import ps_roi_pool, PSRoIPool from .ps_roi_pool import ps_roi_pool, PSRoIPool
...@@ -51,7 +51,6 @@ __all__ = [ ...@@ -51,7 +51,6 @@ __all__ = [
"stochastic_depth", "stochastic_depth",
"StochasticDepth", "StochasticDepth",
"FrozenBatchNorm2d", "FrozenBatchNorm2d",
"ConvNormActivation",
"SqueezeExcitation", "SqueezeExcitation",
"generalized_box_iou_loss", "generalized_box_iou_loss",
] ]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment