gigl.experimental.knowledge_graph_embedding.lib.config.run

Classes

RunConfig

Configuration for runtime execution environment settings.

Module Contents

class gigl.experimental.knowledge_graph_embedding.lib.config.run.RunConfig

Configuration for runtime execution environment settings.

Controls the basic execution environment for knowledge graph embedding training, particularly hardware acceleration preferences.

should_use_cuda

Whether to use CUDA (GPU) acceleration for training. If True, training will use available GPUs for faster computation. If False, training will run on CPU only. Automatically adjusted based on GPU availability during initialization. Defaults to True.

Type:

bool

should_use_cuda: bool = True