diff options
Diffstat (limited to 'data/csv.py')
| -rw-r--r-- | data/csv.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/data/csv.py b/data/csv.py index 253ce9e..aad970c 100644 --- a/data/csv.py +++ b/data/csv.py | |||
| @@ -23,7 +23,7 @@ class CSVDataModule(pl.LightningDataModule): | |||
| 23 | tokenizer, | 23 | tokenizer, |
| 24 | instance_identifier, | 24 | instance_identifier, |
| 25 | class_identifier=None, | 25 | class_identifier=None, |
| 26 | class_subdir="db_cls", | 26 | class_subdir="cls", |
| 27 | num_class_images=100, | 27 | num_class_images=100, |
| 28 | size=512, | 28 | size=512, |
| 29 | repeats=100, | 29 | repeats=100, |