File size: 868 Bytes
3be620b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
model:
checkpoint_path: ../../../checkpoints/kny_image_full_no_disc/checkpoint
vqvae_config:
beta: 0.25
num_embeddings: 50257
embedding_dim: 128
autoencoder_config:
z_channels: 512
channels: 32
channels_multiplier:
- 2
- 4
- 8
- 8
num_res_blocks: 1
attention_resolution:
- 16
resolution: 128
dropout: 0.0
discriminator_config:
num_layers: 3
filters: 64
loss_config:
discriminator:
loss: "hinge"
factor: 1.0
iter_start: 5000
weight: 0.8
vqvae:
codebook_weight: 1.0
perceptual_weight: 4.0
perceptual_loss: "vgg19" # "vgg16", "vgg19", "style"
trainer:
batch_size: 32
n_epochs: 10000
gen_lr: 3e-5
disc_lr: 3e-5
gen_beta_1: 0.5
gen_beta_2: 0.9
disc_beta_1: 0.5
disc_beta_2: 0.9
gen_clip_norm: 1.0
disc_clip_norm: 1.0
|