peal.generators.rae_pipeline.load_generator_config

peal.generators.rae_pipeline.load_generator_config(path)

Load an RAE generator yaml and fill in the pipeline defaults.

Path-valued keys (base_path, stage1_config, stage2_config, raev2_dir, data) are expanded, and defaults are set for raev2_dir, the experiment names, precisions, stats sample counts, keep_checkpoints and the stage-2 cache settings.

Parameters:

path (str) – Path of the generator config yaml.

Returns:

The config dict consumed by RAEPipeline.

Return type:

dict