Unverified Commit 5cad35bd authored by Ryan Li's avatar Ryan Li Committed by GitHub
Browse files

fix deprecated wrappers exiting bug (#567)

parent cc332b26
...@@ -7,7 +7,7 @@ from ..cnn.bricks.wrappers import Conv2d, ConvTranspose2d, Linear, MaxPool2d ...@@ -7,7 +7,7 @@ from ..cnn.bricks.wrappers import Conv2d, ConvTranspose2d, Linear, MaxPool2d
class Conv2d_deprecated(Conv2d): class Conv2d_deprecated(Conv2d):
def __init__(*args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
warnings.warn( warnings.warn(
'Importing Conv2d wrapper from "mmcv.ops" will be deprecated in' 'Importing Conv2d wrapper from "mmcv.ops" will be deprecated in'
...@@ -16,7 +16,7 @@ class Conv2d_deprecated(Conv2d): ...@@ -16,7 +16,7 @@ class Conv2d_deprecated(Conv2d):
class ConvTranspose2d_deprecated(ConvTranspose2d): class ConvTranspose2d_deprecated(ConvTranspose2d):
def __init__(*args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
warnings.warn( warnings.warn(
'Importing ConvTranspose2d wrapper from "mmcv.ops" will be ' 'Importing ConvTranspose2d wrapper from "mmcv.ops" will be '
...@@ -26,7 +26,7 @@ class ConvTranspose2d_deprecated(ConvTranspose2d): ...@@ -26,7 +26,7 @@ class ConvTranspose2d_deprecated(ConvTranspose2d):
class MaxPool2d_deprecated(MaxPool2d): class MaxPool2d_deprecated(MaxPool2d):
def __init__(*args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
warnings.warn( warnings.warn(
'Importing MaxPool2d wrapper from "mmcv.ops" will be deprecated in' 'Importing MaxPool2d wrapper from "mmcv.ops" will be deprecated in'
...@@ -35,7 +35,7 @@ class MaxPool2d_deprecated(MaxPool2d): ...@@ -35,7 +35,7 @@ class MaxPool2d_deprecated(MaxPool2d):
class Linear_deprecated(Linear): class Linear_deprecated(Linear):
def __init__(*args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
warnings.warn( warnings.warn(
'Importing Linear wrapper from "mmcv.ops" will be deprecated in' 'Importing Linear wrapper from "mmcv.ops" will be deprecated in'
......
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