stylegan2/__init__.py

from .generator import SynthesisNetwork
from .generator import MappingNetwork
from .generator import Generator
from .discriminator import Discriminator