diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-26 00:35:02 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-26 00:35:02 +0200 |
| commit | 22e36513dd43d2e2ca82ca28a1ea757c5663676a (patch) | |
| tree | 54285c3c30a02b00af989078bf61c122b9eccabd /text_recognizer/data/emnist.py | |
| parent | 9c3a8753d95ecb70a84e1eb40933590a510abfc4 (diff) | |
Updates
Diffstat (limited to 'text_recognizer/data/emnist.py')
| -rw-r--r-- | text_recognizer/data/emnist.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/text_recognizer/data/emnist.py b/text_recognizer/data/emnist.py index bf3faec..824b947 100644 --- a/text_recognizer/data/emnist.py +++ b/text_recognizer/data/emnist.py @@ -10,7 +10,7 @@ import h5py from loguru import logger import numpy as np import toml -from torchvision import transforms +import torchvision.transforms as T from text_recognizer.data.base_data_module import ( BaseDataModule, @@ -53,7 +53,7 @@ class EMNIST(BaseDataModule): self.data_train = None self.data_val = None self.data_test = None - self.transform = transforms.Compose([transforms.ToTensor()]) + self.transform = T.Compose([T.ToTensor()]) self.dims = (1, *self.input_shape) self.output_dims = (1,) |