blob: 34a67aefc556cf043893bf25db431f3b77204e18 (
plain)
1
2
3
4
5
6
7
8
9
10
11
|
_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
|