def _get_scheduler(
*, optimizer: Union[Optimizer, Dict[str, Optimizer]],
**params: Any) -> Union[Scheduler, Dict[str, Scheduler]]:
optimizer_key = params.pop("_optimizer", None)
optimizer = optimizer[optimizer_key] if optimizer_key else optimizer
scheduler = SCHEDULERS.get_from_params(**params, optimizer=optimizer)
return scheduler
def _get_scheduler(*, optimizer, **params):
key_value_flag = params.pop("_key_value", False)
if key_value_flag:
scheduler = {}
for scheduler_key, scheduler_params in params.items():
scheduler[
scheduler_key] = ConfigExperiment._get_scheduler( # noqa: WPS437
optimizer=optimizer,
**scheduler_params)
else:
scheduler = SCHEDULERS.get_from_params(**params,
optimizer=optimizer)
return scheduler