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