allennlp.training.optimizers

AllenNLP just uses PyTorch optimizers , with a thin wrapper to allow registering them and instantiating them from_params.

The available optimizers are

class allennlp.training.optimizers.Optimizer[source]

Bases: allennlp.common.registrable.Registrable

This class just allows us to implement Registrable for Pytorch Optimizers.

default_implementation = 'adam'
classmethod from_params(model_parameters: typing.List[torch.nn.parameter.Parameter], params: allennlp.common.params.Params)[source]