Get the device of this module. Will prioritize the module’s setting over the global one.
BaseDataset.device
dgs.models.dataset.dataset.BaseDataset.validate_params
dgs.models.dataset.dataset.BaseDataset.is_training