From a06aed4a99e2caf21e2997ab5314caf2bf7c8a64 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 17 Nov 2021 22:39:33 +0100 Subject: Fix import path vqganloss --- text_recognizer/criterion/vqgan_loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'text_recognizer/criterion') 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( -- cgit v1.2.3-70-g09d2