diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-27 22:42:10 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-27 22:42:10 +0200 |
commit | 1a686963adb57118d5641adebf5185bfacaea015 (patch) | |
tree | 8914433e880ecf9a02a5143dd1f3f1a4215f729b /text_recognizer | |
parent | ce5e045a271ee497b3614364f8b476b8d52a24c5 (diff) |
Fix import of criterion in VQGAN
Diffstat (limited to 'text_recognizer')
-rw-r--r-- | text_recognizer/models/vqgan.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/models/vqgan.py b/text_recognizer/models/vqgan.py index 6e4d3fb..6a90e06 100644 --- a/text_recognizer/models/vqgan.py +++ b/text_recognizer/models/vqgan.py @@ -4,8 +4,8 @@ from typing import Tuple import attr from torch import Tensor +from text_recognizer.criterion.vqgan_loss import VQGANLoss from text_recognizer.models.base import BaseLitModel -from text_recognizer.criterions.vqgan_loss import VQGANLoss @attr.s(auto_attribs=True, eq=False) |