Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
POSTECH-CVLab
GitHub Repository: POSTECH-CVLab/PyTorch-StudioGAN
Path: blob/master/src/configs/ImageNet/StyleGAN3-t-SPD.yaml
809 views
1
DATA:
2
name: "ImageNet"
3
img_size: 128
4
num_classes: 1000
5
MODEL:
6
backbone: "stylegan3"
7
g_cond_mtd: "cAdaIN"
8
d_cond_mtd: "SPD"
9
g_act_fn: "Auto"
10
d_act_fn: "Auto"
11
z_prior: "gaussian"
12
z_dim: 512
13
w_dim: 512
14
g_conv_dim: "N/A"
15
d_conv_dim: "N/A"
16
apply_g_ema: True
17
LOSS:
18
adv_loss: "logistic"
19
apply_r1_reg: True
20
r1_lambda: 0.5
21
r1_place: "outside_loop"
22
OPTIMIZATION:
23
# These values will be slightly changed if lazy regularization is applied.
24
batch_size: 256
25
acml_steps: 1
26
g_lr: 0.0025
27
d_lr: 0.002
28
beta1: 0
29
beta2: 0.99
30
d_first: False
31
g_updates_per_step: 1
32
d_updates_per_step: 1
33
total_steps: 200000
34
STYLEGAN:
35
stylegan3_cfg: "stylegan3-t"
36
g_reg_interval: 1
37
d_reg_interval: 16
38
mapping_network: 2
39
style_mixing_p: 0
40
g_ema_kimg: 10
41
g_ema_rampup: 0.05
42
apply_pl_reg: False
43
pl_weight: 0
44
d_architecture: "resnet"
45
d_epilogue_mbstd_group_size: 4
46
47