Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
POSTECH-CVLab
GitHub Repository: POSTECH-CVLab/PyTorch-StudioGAN
 Path: blob/master/src/configs/CUB200/StyleGAN2-D2DCE-ADA.yaml
809 views
1
DATA:

2
  name: "CUB200"

3
  img_size: 128

4
  num_classes: 200

5
MODEL:

6
  backbone: "stylegan2"

7
  g_cond_mtd: "cAdaIN"

8
  d_cond_mtd: "D2DCE"

9
  normalize_d_embed: True

10
  d_embed_dim: 1024

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.95

23
  temperature: 0.25

24
  apply_r1_reg: True

25
  r1_lambda: 0.2048

26
  r1_place: "outside_loop"

27
OPTIMIZATION:

28
  # These values will be slightly changed if lazy regularization is applied.

29
  batch_size: 64

30
  acml_steps: 1

31
  g_lr: 0.0025

32
  d_lr: 0.0025

33
  beta1: 0

34
  beta2: 0.99

35
  d_first: False

36
  g_updates_per_step: 1

37
  d_updates_per_step: 2

38
  total_steps: 100000

39
AUG:

40
  apply_ada: True

41
  ada_aug_type: "bgc"

42
  ada_initial_augment_p: 0

43
  ada_target: 0.6

44
  ada_kimg: 500

45
  ada_interval: 4

46
STYLEGAN:

47
  g_reg_interval: 4

48
  d_reg_interval: 16

49
  mapping_network: 2

50
  style_mixing_p: 0.9

51
  g_ema_kimg: 20

52
  g_ema_rampup: "N/A"

53
  apply_pl_reg: True

54
  pl_weight: 2

55
  d_architecture: "resnet"

56
  d_epilogue_mbstd_group_size: 8

57


58