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):
def __init__(self, split_filename: str):
montgomery_loader = MontgomeryLoader()
montgomery_split = make_montgomery_split("default.json")
shenzen_loader = ShenzhenLoader()
shenzen_split = make_shenzhen_split("default.json")
shenzhen_loader = ShenzhenLoader()
shenzhen_split = make_shenzhen_split("default.json")
super().__init__(
splits={
"train": [
(montgomery_split["train"], montgomery_loader),
(shenzen_split["train"], shenzen_loader),
(shenzhen_split["train"], shenzhen_loader),
],
"validation": [
(montgomery_split["validation"], montgomery_loader),
(shenzen_split["validation"], shenzen_loader),
(shenzhen_split["validation"], shenzhen_loader),
],
"test": [
(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