sketchgraphs_models.graph.train

Main module for training entity model.

Functions

sketchgraphs_models.graph.train.get_argsparser()

sketchgraphs_models.graph.train.main()

Default main function.

sketchgraphs_models.graph.train.make_model_with_arguments(feature_dimensions, args)

sketchgraphs_models.graph.train.run(args, distributed_config=None)

Runs the entire training process according to the given configuration.

sketchgraphs_models.graph.train.train(node_feature_mapping, edge_feature_mapping, dataloader_train, args, output_dir=None, dataloader_eval=None, batches_per_epoch=None, dist_config=None)

Modules

sketchgraphs_models.graph.train.data_loading	This module contains the main functions used to load the required data from disk for training.
sketchgraphs_models.graph.train.harness	This module contains the main training harness for training graph models.