Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
POSTECH-CVLab
GitHub Repository: POSTECH-CVLab/PyTorch-StudioGAN
Path: blob/master/src/configs/CIFAR100/StyleGAN2-D2DCE-ADA.yaml
809 views
1
DATA:
2
name: "CIFAR100"
3
img_size: 32
4
num_classes: 100
5
MODEL:
6
backbone: "stylegan2"
7
g_cond_mtd: "cAdaIN"
8
d_cond_mtd: "D2DCE"
9
normalize_d_embed: True
10
d_embed_dim: 512
11
g_act_fn: "Auto"
12
d_act_fn: "Auto"
13
z_prior: "gaussian"
14
z_dim: 512
15
w_dim: 512
16
g_conv_dim: "N/A"
17
d_conv_dim: "N/A"
18
apply_g_ema: True
19
LOSS:
20
adv_loss: "logistic"
21
cond_lambda: 0.25
22
m_p: 0.98
23
temperature: 0.25
24
apply_r1_reg: True
25
r1_lambda: 0.01
26
r1_place: "outside_loop"
27
OPTIMIZATION:
28
# These values will be slightly changed if lazy regularization is applied.
29
g_lr: 0.0025
30
d_lr: 0.0025
31
beta1: 0
32
beta2: 0.99
33
d_first: False
34
g_updates_per_step: 1
35
d_updates_per_step: 2
36
total_steps: 1000000
37
AUG:
38
apply_ada: True
39
ada_aug_type: "bgc"
40
ada_initial_augment_p: 0
41
ada_target: 0.8
42
ada_kimg: 500
43
ada_interval: 4
44
STYLEGAN:
45
g_reg_interval: 4
46
d_reg_interval: 16
47
mapping_network: 2
48
style_mixing_p: 0
49
g_ema_kimg: 500
50
g_ema_rampup: 0.05
51
apply_pl_reg: False
52
pl_weight: 0
53
d_architecture: "orig"
54
d_epilogue_mbstd_group_size: 32
55
56