sketchgraphs_models.autoconstraint.train

Main module for training auto-constraint model.

Classes

AutoconstraintHarness(model, opt, …[, …])

Functions

sketchgraphs_models.autoconstraint.train.get_argsparser()

sketchgraphs_models.autoconstraint.train.initialize_datasets(args, distributed_config)

sketchgraphs_models.autoconstraint.train.main()

Default main function.

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

Runs the entire training process according to the given configuration.

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