File size: 763 Bytes
8c212a5 |
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 |
# python3.7
"""Configuration for testing StyleGAN on FF-HQ (1024) dataset.
All settings are particularly used for one replica (GPU), such as `batch_size`
and `num_workers`.
"""
runner_type = 'StyleGANRunner'
gan_type = 'stylegan'
resolution = 1024
batch_size = 16
data = dict(
num_workers=4,
# val=dict(root_dir='data/ffhq', resolution=resolution),
val=dict(root_dir='data/ffhq.zip', data_format='zip',
resolution=resolution),
)
modules = dict(
discriminator=dict(
model=dict(gan_type=gan_type, resolution=resolution),
kwargs_val=dict(),
),
generator=dict(
model=dict(gan_type=gan_type, resolution=resolution),
kwargs_val=dict(trunc_psi=0.7, trunc_layers=8, randomize_noise=False),
)
)
|