summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--text_recognizer/criterion/vqgan_loss.py2
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(