Spaces:
Runtime error
Runtime error
File size: 881 Bytes
1ba3df3 |
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 |
models:
G:
encoder:
content:
type: ContentVanillaEncoder
depth: 2
style:
type: StyleVanillaEncoder
depth: 2
decoder:
type: VanillaDecoder
residual_blocks: 6
depth: 2
optim:
class: torch.optim.Adam
betas: [ 0.5, 0.999 ]
lr: 0.0002
lr_policy: step
lr_decay_iters: 1000
init_type: normal
init_gain: 0.02
D_content:
in_channels: 2 # char + 1
class: models.discriminator.PatchGANDiscriminator
optim:
class: torch.optim.Adam
betas: [ 0.5, 0.999 ]
lr: 2e-4
lr_policy: step
lr_decay_iters: 1000
D_style:
in_channels: 6 # style + 1
class: models.discriminator.PatchGANDiscriminator
optim:
class: torch.optim.Adam
betas: [ 0.5, 0.999 ]
lr: 2e-4
lr_policy: step
lr_decay_iters: 1000 |