Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
POSTECH-CVLab
GitHub Repository: POSTECH-CVLab/PyTorch-StudioGAN
Path: blob/master/src/configs/CUB200/ACGAN-Mod-Big-ADC.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: "AC"
9
aux_cls_type: "ADC"
10
apply_g_sn: True
11
apply_d_sn: True
12
apply_attn: True
13
attn_g_loc: [4]
14
attn_d_loc: [1]
15
z_dim: 120
16
g_shared_dim: 128
17
g_conv_dim: 80
18
d_conv_dim: 80
19
apply_g_ema: True
20
g_ema_decay: 0.9999
21
g_ema_start: 4000
22
LOSS:
23
adv_loss: "hinge"
24
cond_lambda: 1.0
25
OPTIMIZATION:
26
batch_size: 256
27
g_lr: 0.00005
28
d_lr: 0.0002
29
beta1: 0.0
30
beta2: 0.999
31
d_updates_per_step: 2
32
total_steps: 40000
33
34