config

Arg Descriptions

Config Dict

diagnnose.config.config_dict.create_config_dict() Dict[str, Dict[str, Any]][source]

Sets up the configuration for extraction.

Config can be provided from a json file or the commandline. Values in the json file can be overwritten by providing them from the commandline. Will raise an error if a required argument is not provided in either the json file or as a commandline arg.

Commandline args should be provided as dot-separated values, where the first dot indicates the arg group the arg belongs to. For example, setting the state_dict of a model can be done with the flag --model.state_dict state_dict.

Returns

config_dict – Dictionary mapping each arg group to their config values.

Return type

ConfigDict