nat.profiler.parameter_optimization.parameter_optimizer#

Attributes#

logger

Functions#

optimize_parameters(→ nat.data_models.config.Config)

Tune all non-prompt hyper-parameters and persist the best config.

Module Contents#

logger#
optimize_parameters(
*,
base_cfg: nat.data_models.config.Config,
full_space: collections.abc.Mapping[str, nat.data_models.optimizable.SearchSpace],
optimizer_config: nat.data_models.optimizer.OptimizerConfig,
opt_run_config: nat.data_models.optimizer.OptimizerRunConfig,
) → nat.data_models.config.Config#

Tune all non-prompt hyper-parameters and persist the best config.