jesterTOV.inference.flows.flow.load_model

load_model(output_dir)

Load a trained flow model from saved files.

Parameters:

output_dir (str) – Directory containing saved model files

Return type:

Tuple[Any, Dict[str, Any]]

Returns:

flow – Loaded flow model metadata: Training metadata (includes data statistics if standardization was used)

Example

>>> flow, metadata = load_model("./models/gw170817/")