core.utils.torchutils.get_scheduler_from_kwargs

core.utils.torchutils.get_scheduler_from_kwargs(optimizer, **scheduler_kwargs)

Builds and returns an scheduler for optimizer. The type of the scheduler is determined by kwarg type, the remaining kwargs are passed to the scheduler.

Parameters

Name	Type	Description	Default
optimizer	torch.optim.Optimizer	optimizer for which the scheduler is used	required
scheduler_kwargs		kwargs for scheduler; type needs to be one of [step, cosine, reduce_on_plateau], the remaining kwargs are used for specific scheduler kwargs, such as learning rate and momentum	{}

Returns

Name	Type	Description
	scheduler