gigl.src.training.v1.trainer#

Attributes#

logger

parser

Classes#

Trainer

GiGL Component that trains a GNN model using the specified task and resource configurations.

Module Contents#

class gigl.src.training.v1.trainer.Trainer[source]#

GiGL Component that trains a GNN model using the specified task and resource configurations.

run(applied_task_identifier, task_config_uri, resource_config_uri, cpu_docker_uri=None, cuda_docker_uri=None)[source]#
Parameters:

  • applied_task_identifier (gigl.src.common.types.AppliedTaskIdentifier)

  • task_config_uri (gigl.common.Uri)

  • resource_config_uri (gigl.common.Uri)

  • cpu_docker_uri (Optional[str])

  • cuda_docker_uri (Optional[str])

Return type:

None

gigl.src.training.v1.trainer.logger[source]#
gigl.src.training.v1.trainer.parser[source]#