Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
POSTECH-CVLab
GitHub Repository: POSTECH-CVLab/PyTorch-StudioGAN
 Path: blob/master/src/configs/AFHQv2/StyleGAN3-r-paper.yaml
809 views
1
# Requires AFHQ dataset folders to be combined into 1 class

2
DATA:

3
  name: "AFHQv2"

4
  img_size: 512

5
  num_classes: 3

6
MODEL:

7
  backbone: "stylegan3"

8
  g_cond_mtd: "W/O"

9
  d_cond_mtd: "W/O"

10
  g_act_fn: "Auto"

11
  d_act_fn: "Auto"

12
  z_prior: "gaussian"

13
  z_dim: 512

14
  w_dim: 512

15
  g_conv_dim: "N/A"

16
  d_conv_dim: "N/A"

17
  apply_g_ema: True

18
LOSS:

19
  adv_loss: "logistic"

20
  apply_r1_reg: True

21
  r1_lambda: 16.4

22
  r1_place: "outside_loop"

23
OPTIMIZATION:

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

25
  batch_size: 32

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

34
AUG:

35
  apply_ada: True

36
  ada_aug_type: "bgc"

37
  ada_initial_augment_p: 0

38
  ada_target: 0.6

39
  ada_kimg: 500

40
  ada_interval: 4

41
STYLEGAN:

42
  stylegan3_cfg: "stylegan3-r"

43
  blur_init_sigma: 10

44
  g_reg_interval: 1

45
  d_reg_interval: 16

46
  mapping_network: 2

47
  style_mixing_p: 0

48
  g_ema_kimg: 10

49
  g_ema_rampup: 0.05

50
  apply_pl_reg: False

51
  pl_weight: 0

52
  d_architecture: "resnet"

53
  d_epilogue_mbstd_group_size: 4

54


55