Fixed CTC loss.

1 files changed, 1 insertions, 8 deletions

diff --git a/src/text_recognizer/datasets/emnist_dataset.py b/src/text_recognizer/datasets/emnist_dataset.py
index 81268fb..d01dcee 100644
--- a/src/text_recognizer/datasets/emnist_dataset.py
+++ b/src/text_recognizer/datasets/emnist_dataset.py
@@ -13,17 +13,10 @@ from torchvision.datasets import EMNIST
 from torchvision.transforms import Compose, ToTensor
 
 from text_recognizer.datasets.dataset import Dataset
+from text_recognizer.datasets.transforms import Transpose
 from text_recognizer.datasets.util import DATA_DIRNAME
 
 
-class Transpose:
-    """Transposes the EMNIST image to the correct orientation."""
-
-    def __call__(self, image: Image) -> np.ndarray:
-        """Swaps axis."""
-        return np.array(image).swapaxes(0, 1)
-
-
 class EmnistDataset(Dataset):
     """This is a class for resampling and subsampling the PyTorch EMNIST dataset."""