From e5eca28438cd17d436359f2c6eee0bb9e55d2a8b Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 2 Aug 2021 01:01:51 +0200 Subject: Fixed wandb callback bug, add word_pieces arg to iam paragraph --- text_recognizer/data/mappings.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'text_recognizer/data/mappings.py') diff --git a/text_recognizer/data/mappings.py b/text_recognizer/data/mappings.py index a934fd9..b69e888 100644 --- a/text_recognizer/data/mappings.py +++ b/text_recognizer/data/mappings.py @@ -4,7 +4,7 @@ from pathlib import Path from typing import Dict, List, Optional, Union, Set, Sequence import attr -from loguru import logger +import loguru.logger as log import torch from torch import Tensor @@ -87,7 +87,7 @@ class WordPieceMapping(EmnistMapping): if self.data_dir is None else Path(self.data_dir) ) - logger.debug(f"Using data dir: {self.data_dir}") + log.debug(f"Using data dir: {self.data_dir}") if not self.data_dir.exists(): raise RuntimeError(f"Could not locate iamdb directory at {self.data_dir}") -- cgit v1.2.3-70-g09d2