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

[montgomery_shenzhen] Fix use of split filename

parent 9f75a244
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
......@@ -14,9 +14,9 @@ class DataModule(ConcatDataModule):
def __init__(self, split_filename: str):
montgomery_loader = MontgomeryLoader()
montgomery_split = make_montgomery_split("default.json")
montgomery_split = make_montgomery_split(split_filename)
shenzhen_loader = ShenzhenLoader()
shenzhen_split = make_shenzhen_split("default.json")
shenzhen_split = make_shenzhen_split(split_filename)
super().__init__(
splits={
......
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