gigl.src.training.v1.lib.data_loaders.common#

Attributes#

logger

Classes#

DataloaderConfig

DataloaderTypes

Generic enumeration.

Module Contents#

class gigl.src.training.v1.lib.data_loaders.common.DataloaderConfig[source]#
batch_size: int = 32[source]#
num_workers: int = 0[source]#
pin_memory: bool = False[source]#
seed: int = 42[source]#
should_loop: bool = False[source]#
uris: list[gigl.common.Uri] | dict[gigl.src.common.types.graph_data.NodeType, list[gigl.common.Uri]][source]#
class gigl.src.training.v1.lib.data_loaders.common.DataloaderTypes[source]#

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

test_main = 'test_main'[source]#
test_random_negative = 'test_random_negative'[source]#
train_main = 'train_main'[source]#
train_random_negative = 'train_random_negative'[source]#
val_main = 'val_main'[source]#
val_random_negative = 'val_random_negative'[source]#
gigl.src.training.v1.lib.data_loaders.common.logger[source]#