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-ADA.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
OPTIMIZATION:

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

24
  batch_size: 64

25
  acml_steps: 1

26
  g_lr: 0.0025

27
  d_lr: 0.0025

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
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
  g_reg_interval: 4

43
  d_reg_interval: 16

44
  mapping_network: 8

45
  style_mixing_p: 0.9

46
  g_ema_kimg: 20

47
  g_ema_rampup: "N/A"

48
  apply_pl_reg: True

49
  pl_weight: 2

50
  d_architecture: "resnet"

51
  d_epilogue_mbstd_group_size: 8

52


53