diff options
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 |