Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
POSTECH-CVLab
GitHub Repository: POSTECH-CVLab/PyTorch-StudioGAN
Path: blob/master/src/configs/CUB200/BigGAN.yaml
809 views
1
DATA:
2
name: "CUB200"
3
img_size: 128
4
num_classes: 200
5
MODEL:
6
backbone: "big_resnet"
7
g_cond_mtd: "cBN"
8
d_cond_mtd: "PD"
9
apply_g_sn: True
10
apply_d_sn: True
11
apply_attn: True
12
attn_g_loc: [4]
13
attn_d_loc: [1]
14
z_dim: 120
15
g_shared_dim: 128
16
g_conv_dim: 80
17
d_conv_dim: 80
18
apply_g_ema: True
19
g_ema_decay: 0.9999
20
g_ema_start: 4000
21
LOSS:
22
adv_loss: "hinge"
23
OPTIMIZATION:
24
batch_size: 256
25
g_lr: 0.00005
26
d_lr: 0.0002
27
beta1: 0.0
28
beta2: 0.999
29
d_updates_per_step: 2
30
total_steps: 40000
31
32