diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-15 21:15:31 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-15 21:15:31 +0200 |
commit | da7d2171c818afefb3bad3cd66ce85fddd519c1c (patch) | |
tree | bc2cd9f2aeca62cc2793a6882ee96ab5033868e2 /training/conf/experiment/vqgan.yaml | |
parent | 441b7484348953deb7c94150675d54583ef5a81a (diff) |
Updates to VQGAN loss
Diffstat (limited to 'training/conf/experiment/vqgan.yaml')
-rw-r--r-- | training/conf/experiment/vqgan.yaml | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/training/conf/experiment/vqgan.yaml b/training/conf/experiment/vqgan.yaml index 554ec9e..40af15a 100644 --- a/training/conf/experiment/vqgan.yaml +++ b/training/conf/experiment/vqgan.yaml @@ -19,7 +19,9 @@ criterion: num_channels: 32 num_layers: 3 vq_loss_weight: 0.8 - discriminator_weight: 0.6 + discriminator_weight: 0.8 + discriminator_factor: 1.0 + discriminator_iter_start: 2e4 datamodule: batch_size: 8 |