trainer

class fairdiverse.search.trainer.SRDTrainer(train_config)

Bases: object

load_configs(dir)

Loads and merges configuration files for the model, dataset, and evaluation.

This function loads multiple YAML configuration files, including the process configuration, dataset-specific settings, model configurations, and evaluation parameters. All configurations are merged, with the highest priority given to the class’s own config attribute.

Parameters:

dir – The directory where the main process configuration file is located.

Returns:

A dictionary containing the merged configuration from all files.

train()

Training post-processing search model main workflow.