diff options
| -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( |