__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .styleganv2_modules import (Blur, ConstantInput, ModulatedConv2d,
                                 ModulatedStyleConv, ModulatedToRGB,
                                 NoiseInjection)
from .styleganv3_modules import (MappingNetwork, SynthesisInput,
                                 SynthesisLayer, SynthesisNetwork)

__all__ = [
    'Blur', 'ModulatedStyleConv', 'ModulatedToRGB', 'NoiseInjection',
    'ConstantInput', 'MappingNetwork', 'SynthesisInput', 'SynthesisLayer',
    'SynthesisNetwork', 'ModulatedConv2d'
]