diff options
Diffstat (limited to 'training/conf/criterion/vqgan_loss.yaml')
| -rw-r--r-- | training/conf/criterion/vqgan_loss.yaml | 11 |
1 files changed, 0 insertions, 11 deletions
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 |