GraphDataLoader

class caldera.data.GraphDataLoader(*dataset, batch_size=1, shuffle=False, **kwargs)

Parameters

• dataset (Union[List[GraphBatch], Dataset]) –

• batch_size (int) –

• shuffle (bool) –

__init__(*dataset, batch_size=1, shuffle=False, **kwargs)

Parameters

• dataset (Union[List[GraphBatch], Dataset]) –

• batch_size (int) –

• shuffle (bool) –

Methods

__init__(*dataset[, batch_size, shuffle])	param dataset
first(*args, **kwargs)
mem_sizes()

Attributes

multiprocessing_context