- 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