diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-10 23:25:13 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-10 23:25:13 +0200 |
| commit | 441b7484348953deb7c94150675d54583ef5a81a (patch) | |
| tree | 6a3e537f98a5c34fe166db7f4c60552b1212b8f5 /training/conf/experiment/vqgan.yaml | |
| parent | 82f4acabe24e5171c40afa2939a4777ba87bcc30 (diff) | |
Update to config and logging in VQGAN
Diffstat (limited to 'training/conf/experiment/vqgan.yaml')
| -rw-r--r-- | training/conf/experiment/vqgan.yaml | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/training/conf/experiment/vqgan.yaml b/training/conf/experiment/vqgan.yaml index 570e7f9..554ec9e 100644 --- a/training/conf/experiment/vqgan.yaml +++ b/training/conf/experiment/vqgan.yaml @@ -8,6 +8,19 @@ defaults: - override /optimizers: null - override /lr_schedulers: null +criterion: + _target_: text_recognizer.criterions.vqgan_loss.VQGANLoss + reconstruction_loss: + _target_: torch.nn.L1Loss + reduction: mean + discriminator: + _target_: text_recognizer.criterions.n_layer_discriminator.NLayerDiscriminator + in_channels: 1 + num_channels: 32 + num_layers: 3 + vq_loss_weight: 0.8 + discriminator_weight: 0.6 + datamodule: batch_size: 8 @@ -33,7 +46,7 @@ lr_schedulers: optimizers: generator: _target_: madgrad.MADGRAD - lr: 2.0e-5 + lr: 4.5e-6 momentum: 0.5 weight_decay: 0 eps: 1.0e-6 @@ -42,7 +55,7 @@ optimizers: discriminator: _target_: madgrad.MADGRAD - lr: 2.0e-5 + lr: 4.5e-6 momentum: 0.5 weight_decay: 0 eps: 1.0e-6 |