fit_a_nef.utils.get_meta_init

fit_a_nef.utils.get_meta_init(storage_folder: Path, epoch_index: int) → Array | None

Load meta-learned initialization for current configuration. TODO this is not implemented properly yet.

Parameters:

• storage_folder (Path) – The folder to load the meta-learned initialization from.

• epoch_index (int) – The epoch index to load.

Returns:

The meta-learned initialization.

Return type:

Union[jnp.ndarray, None]