summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-17 22:39:33 +0100
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-17 22:39:33 +0100
commita06aed4a99e2caf21e2997ab5314caf2bf7c8a64 (patch)
treef89cc5c6b07884dbcd1ab4d8ce3c7882f9b7996d
parent6ff1444dc46153e7d0e5756c46b3e14a4f353d63 (diff)
Fix import path vqganloss
-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(