Skip to content
Snippets Groups Projects
Commit 9f75a244 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[montgomery_shenzhen] Fix typo in naming

parent b280ec3a
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
...@@ -15,22 +15,22 @@ class DataModule(ConcatDataModule): ...@@ -15,22 +15,22 @@ class DataModule(ConcatDataModule):
def __init__(self, split_filename: str): def __init__(self, split_filename: str):
montgomery_loader = MontgomeryLoader() montgomery_loader = MontgomeryLoader()
montgomery_split = make_montgomery_split("default.json") montgomery_split = make_montgomery_split("default.json")
shenzen_loader = ShenzhenLoader() shenzhen_loader = ShenzhenLoader()
shenzen_split = make_shenzhen_split("default.json") shenzhen_split = make_shenzhen_split("default.json")
super().__init__( super().__init__(
splits={ splits={
"train": [ "train": [
(montgomery_split["train"], montgomery_loader), (montgomery_split["train"], montgomery_loader),
(shenzen_split["train"], shenzen_loader), (shenzhen_split["train"], shenzhen_loader),
], ],
"validation": [ "validation": [
(montgomery_split["validation"], montgomery_loader), (montgomery_split["validation"], montgomery_loader),
(shenzen_split["validation"], shenzen_loader), (shenzhen_split["validation"], shenzhen_loader),
], ],
"test": [ "test": [
(montgomery_split["test"], montgomery_loader), (montgomery_split["test"], montgomery_loader),
(shenzen_split["test"], shenzen_loader), (shenzhen_split["test"], shenzhen_loader),
], ],
} }
) )
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment