diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-03-23 21:55:42 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-03-23 21:55:42 +0100 |
| commit | ae589fb3ffdbf6c4bb1ae35345f7a3665deeebc5 (patch) | |
| tree | 1702f74c069679ebdd74a03892275c6eb3a80ffd /text_recognizer/datasets/base_data_module.py | |
| parent | e3741de333a3a43a7968241b6eccaaac66dd7b20 (diff) | |
refactored emnist lines dataset
Diffstat (limited to 'text_recognizer/datasets/base_data_module.py')
| -rw-r--r-- | text_recognizer/datasets/base_data_module.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/datasets/base_data_module.py b/text_recognizer/datasets/base_data_module.py index 830b39b..f5e7300 100644 --- a/text_recognizer/datasets/base_data_module.py +++ b/text_recognizer/datasets/base_data_module.py @@ -46,7 +46,7 @@ class BaseDataModule(pl.LightningDataModule): def setup(self, stage: str = None) -> None: """Split into train, val, test, and set dims. - + Should assign `torch Dataset` objects to self.data_train, self.data_val, and optionally self.data_test. |