jesterTOV.inference.flows.train_flow.save_model# save_model(flow, output_dir, flow_kwargs, metadata)[source]# Save trained flow model, architecture kwargs, and metadata. Parameters: flow (Any) – Trained flowjax flow output_dir (str) – Directory to save files flow_kwargs (Dict[str, Any]) – Dictionary of kwargs needed to recreate flow architecture metadata (Dict[str, Any]) – Dictionary with training metadata Return type: None
