Skip to content

dataset.mode in subsets is shared. #100

Description

@gabrieletijunaityte

When splitting datasets, they are just views (subsets) of one original dataset object and thus sharing all parameter, this renders self.mode useless.

if split_data_from_inds:
self.data_train = torch.utils.data.Subset(self.dataset, train_indices)
self.data_train.dataset.mode = "train"
self.data_val = torch.utils.data.Subset(self.dataset, val_indices)
self.data_val.dataset.mode = "val"
if test_indices is not None:
self.data_test = torch.utils.data.Subset(self.dataset, test_indices)
self.data_test.dataset.mode = "test"
else:
self.data_test = None

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions