diff --git a/enhancer/models/complexnn/__init__.py b/enhancer/models/complexnn/__init__.py index 8f9cef6..aa47ad2 100644 --- a/enhancer/models/complexnn/__init__.py +++ b/enhancer/models/complexnn/__init__.py @@ -1 +1,4 @@ from enhancer.models.complexnn.conv import ComplexConv2d # noqa +from enhancer.models.complexnn.conv import ComplexConvTranspose2d # noqa +from enhancer.models.complexnn.rnn import ComplexLSTM # noqa +from enhancer.models.complexnn.utils import ComplexBatchNorm2D # noqa