diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-17 22:39:33 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-17 22:39:33 +0100 |
| commit | a06aed4a99e2caf21e2997ab5314caf2bf7c8a64 (patch) | |
| tree | f89cc5c6b07884dbcd1ab4d8ce3c7882f9b7996d /text_recognizer/criterion | |
| parent | 6ff1444dc46153e7d0e5756c46b3e14a4f353d63 (diff) | |
Fix import path vqganloss
Diffstat (limited to 'text_recognizer/criterion')
| -rw-r--r-- | text_recognizer/criterion/vqgan_loss.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/criterion/vqgan_loss.py b/text_recognizer/criterion/vqgan_loss.py index 9d1cddd..8e8b65b 100644 --- a/text_recognizer/criterion/vqgan_loss.py +++ b/text_recognizer/criterion/vqgan_loss.py @@ -5,7 +5,7 @@ import torch from torch import nn, Tensor import torch.nn.functional as F -from text_recognizer.criterions.n_layer_discriminator import NLayerDiscriminator +from text_recognizer.criterion.n_layer_discriminator import NLayerDiscriminator def _adopt_weight( |