pytorch_optimizer_builder
Module pytorch_optimizer_builder. Specifies a simple factory for building PyTorch optimizers
- pytorch_optimizer_builder.pytorch_optimizer_builder(opt_type: OptimizerType, model_params: Any, **options) Optimizer
Factory method for building PyTorch optimizers
- Parameters
opt_type (The type of the optimizer) –
model_params (Model parameters to optimize on) –
options (Options for the optimizer) –
- Return type
A concrete instance of the optim.Optimizer class