diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-21 21:34:53 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-21 21:34:53 +0100 |
| commit | b44de0e11281c723ec426f8bec8ca0897ecfe3ff (patch) | |
| tree | 998841a3a681d3dedfbe8470c1b8544b4dcbe7a2 /training/conf/criterion | |
| parent | 3b2fb0fd977a6aff4dcf88e1a0f99faac51e05b1 (diff) | |
Remove VQVAE stuff, did not work...
Diffstat (limited to 'training/conf/criterion')
| -rw-r--r-- | training/conf/criterion/mae.yaml | 2 | ||||
| -rw-r--r-- | training/conf/criterion/mse.yaml | 2 | ||||
| -rw-r--r-- | training/conf/criterion/vqgan_loss.yaml | 11 |
3 files changed, 0 insertions, 15 deletions
diff --git a/training/conf/criterion/mae.yaml b/training/conf/criterion/mae.yaml deleted file mode 100644 index cb07467..0000000 --- a/training/conf/criterion/mae.yaml +++ /dev/null @@ -1,2 +0,0 @@ -_target_: torch.nn.L1Loss -reduction: mean diff --git a/training/conf/criterion/mse.yaml b/training/conf/criterion/mse.yaml deleted file mode 100644 index ffd1403..0000000 --- a/training/conf/criterion/mse.yaml +++ /dev/null @@ -1,2 +0,0 @@ -_target_: torch.nn.MSELoss -reduction: mean diff --git a/training/conf/criterion/vqgan_loss.yaml b/training/conf/criterion/vqgan_loss.yaml deleted file mode 100644 index 34a67ae..0000000 --- a/training/conf/criterion/vqgan_loss.yaml +++ /dev/null @@ -1,11 +0,0 @@ -_target_: text_recognizer.criterion.vqgan_loss.VQGANLoss -reconstruction_loss: - _target_: torch.nn.MSELoss - reduction: mean -discriminator: - _target_: text_recognizer.criterions.n_layer_discriminator.NLayerDiscriminator - in_channels: 1 - num_channels: 32 - num_layers: 3 -vq_loss_weight: 1.0 -discriminator_weight: 1.0 |