training/conf/criterion/vqgan_loss.yaml


1
2
3
4
5
6
7
8
9
10
11
12

_target_: text_recognizer.criterions.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