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/vqvae.yaml | |
| parent | 441b7484348953deb7c94150675d54583ef5a81a (diff) | |
Updates to VQGAN loss
Diffstat (limited to 'training/conf/experiment/vqvae.yaml')
| -rw-r--r-- | training/conf/experiment/vqvae.yaml | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/training/conf/experiment/vqvae.yaml b/training/conf/experiment/vqvae.yaml index 8dbb257..d3db471 100644 --- a/training/conf/experiment/vqvae.yaml +++ b/training/conf/experiment/vqvae.yaml @@ -8,6 +8,16 @@ defaults: - override /lr_schedulers: - cosine_annealing +optimizers: + network: + _target_: madgrad.MADGRAD + lr: 3.0e-4 + momentum: 0.9 + weight_decay: 0 + eps: 1.0e-6 + + parameters: network + trainer: max_epochs: 256 |