hookeai.utilities.optimizers

PyTorch-based optimizers.

Functions

get_pytorch_optimizer

Get PyTorch optimizer.

get_learning_rate_scheduler

Get PyTorch optimizer learning rate scheduler.

Functions

get_learning_rate_scheduler(optimizer, ...)	Get PyTorch optimizer learning rate scheduler.
get_pytorch_optimizer(algorithm, params, ...)	Get PyTorch optimizer.