Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
POSTECH-CVLab
GitHub Repository: POSTECH-CVLab/PyTorch-StudioGAN
 Path: blob/master/src/configs/AFHQv2/StyleGAN2-256Res-DiffAug-LeCam.yaml
809 views
1
DATA:

2
  name: "AFHQv2"

3
  img_size: 256

4
  num_classes: 3

5
MODEL:

6
  backbone: "stylegan2"

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: 1

21
  r1_place: "outside_loop"

22
  lecam_lambda: 3e-7

23
  lecam_ema_start_iter: 1000

24
  lecam_ema_decay: 0.99

25
OPTIMIZATION:

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

27
  batch_size: 64

28
  acml_steps: 1

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: 1

36
  total_steps: 200000

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: 8

44
  style_mixing_p: 0.9

45
  g_ema_kimg: 20

46
  g_ema_rampup: "N/A"

47
  apply_pl_reg: True

48
  pl_weight: 2

49
  d_architecture: "resnet"

50
  d_epilogue_mbstd_group_size: 8

51


52