DGLMoleculeDataLoader

class openff.nagl.nn.DGLMoleculeDataLoader(dataset: DGLMoleculeDataset | _LazyDGLMoleculeDataset | ConcatDataset, batch_size: int | None = 1, **kwargs)[source]

Bases: DataLoader

Attributes

dataset

batch_size

num_workers

pin_memory

drop_last

timeout

sampler

pin_memory_device

prefetch_factor