diff --git a/enhancer/models/complexnn/__init__.py b/enhancer/models/complexnn/__init__.py index aa47ad2..918a261 100644 --- a/enhancer/models/complexnn/__init__.py +++ b/enhancer/models/complexnn/__init__.py @@ -2,3 +2,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 +from enhancer.models.complexnn.utils import ComplexRelu # noqa