base_sagan_32x32 源代码

# Copyright (c) OpenMMLab. All rights reserved.
from mmagic.models import DataPreprocessor
from mmagic.models.editors import SAGAN
from mmagic.models.editors.biggan import SelfAttentionBlock
from mmagic.models.editors.sagan import ProjDiscriminator, SNGANGenerator

model = dict(
    type=SAGAN,
    data_preprocessor=dict(type=DataPreprocessor),
    num_classes=10,
    generator=dict(
        type=SNGANGenerator,
        num_classes=10,
        output_scale=32,
        base_channels=256,
        attention_cfg=dict(type=SelfAttentionBlock),
        attention_after_nth_block=2,
        with_spectral_norm=True),
    discriminator=dict(
        type=ProjDiscriminator,
        num_classes=10,
        input_scale=32,
        base_channels=128,
        attention_cfg=dict(type=SelfAttentionBlock),
        attention_after_nth_block=1,
        with_spectral_norm=True),
    generator_steps=1,
    discriminator_steps=5)