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-DiffAug-Rev.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: "inside_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: True

34
  g_updates_per_step: 1

35
  d_updates_per_step: 2

36
  total_steps: 1000000

37
AUG:

38
  apply_diffaug: True

39
  diffaug_type: "diffaug"

40
STYLEGAN:

41
  g_reg_interval: 4

42
  d_reg_interval: 16

43
  mapping_network: 2

44
  style_mixing_p: 0

45
  g_ema_kimg: 500

46
  g_ema_rampup: 0.05

47
  apply_pl_reg: False

48
  pl_weight: 0

49
  d_architecture: "orig"

50
  d_epilogue_mbstd_group_size: 32

51


52