Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
pggan_1024.py 1.52 KB
Newer Older
limm's avatar
add part mmgeneration code  
limm committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58 
# define GAN model
model = dict(
    type='ProgressiveGrowingGAN',
    generator=dict(type='PGGANGenerator', out_scale=1024, noise_size=512),
    discriminator=dict(type='PGGANDiscriminator', in_scale=1024),
    gan_loss=dict(type='GANLoss', gan_type='wgan'),
    disc_auxiliary_loss=[
        dict(
            type='DiscShiftLoss',
            loss_weight=0.001 * 0.5,
            data_info=dict(pred='disc_pred_fake')),
        dict(
            type='DiscShiftLoss',
            loss_weight=0.001 * 0.5,
            data_info=dict(pred='disc_pred_real')),
        dict(
            type='GradientPenaltyLoss',
            loss_weight=10,
            norm_mode='HWC',
            data_info=dict(
                discriminator='disc_partial',
                real_data='real_imgs',
                fake_data='fake_imgs'))
    ])

train_cfg = dict(
    use_ema=True,
    nkimgs_per_scale={
        '4': 600,
        '8': 1200,
        '16': 1200,
        '32': 1200,
        '64': 1200,
        '128': 1200,
        '256': 1200,
        '512': 1200,
        '1024': 12000,
    },
    transition_kimgs=600,
    optimizer_cfg=dict(
        generator=dict(type='Adam', lr=0.001, betas=(0., 0.99)),
        discriminator=dict(type='Adam', lr=0.001, betas=(0., 0.99))),
    g_lr_base=0.001,
    d_lr_base=0.001,
    g_lr_schedule={
        '128': 0.0015,
        '256': 0.002,
        '512': 0.003,
        '1024': 0.003
    },
    d_lr_schedule={
        '128': 0.0015,
        '256': 0.002,
        '512': 0.003,
        '1024': 0.003
    })

test_cfg = None