diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-29 21:40:19 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-29 21:40:19 +0200 |
| commit | 2f1bb639fd5bb6b510af85fb597e9322abc17bc0 (patch) | |
| tree | 3269155b33f33bf2964dc1bdff34d7929b3227f2 /training/conf/criterion | |
| parent | da7d2171c818afefb3bad3cd66ce85fddd519c1c (diff) | |
Remove uploading of code to Wandb, upload config instead
Diffstat (limited to 'training/conf/criterion')
| -rw-r--r-- | training/conf/criterion/vqgan_loss.yaml | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/conf/criterion/vqgan_loss.yaml b/training/conf/criterion/vqgan_loss.yaml index a1c886e..f983f6f 100644 --- a/training/conf/criterion/vqgan_loss.yaml +++ b/training/conf/criterion/vqgan_loss.yaml @@ -1,6 +1,6 @@ _target_: text_recognizer.criterions.vqgan_loss.VQGANLoss reconstruction_loss: - _target_: torch.nn.L1Loss + _target_: torch.nn.MSELoss reduction: mean discriminator: _target_: text_recognizer.criterions.n_layer_discriminator.NLayerDiscriminator |