summaryrefslogtreecommitdiff
path: root/training/conf/criterion/vqgan_loss.yaml
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/criterion/vqgan_loss.yaml')
-rw-r--r--training/conf/criterion/vqgan_loss.yaml11
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