espnet.optimizer package

Initialize sub package.

espnet.optimizer.factory

Import optimizer class dynamically.

class espnet.optimizer.factory.OptimizerFactoryInterface

Bases: object

Optimizer adaptor.

static add_arguments(parser: argparse.ArgumentParser) → argparse.ArgumentParser

Register args.

classmethod build(target, **kwargs)

Initialize optimizer with python-level args.

Parameters:

target – for pytorch model.parameters(), for chainer model

Returns:

new Optimizer

static from_args(target, args: argparse.Namespace)

Initialize optimizer from argparse Namespace.

Parameters:

• target – for pytorch model.parameters(), for chainer model

• args (argparse.Namespace) – parsed command-line args

espnet.optimizer.factory.dynamic_import_optimizer(name: str, backend: str) → espnet.optimizer.factory.OptimizerFactoryInterface

Import optimizer class dynamically.

Parameters:

• name (str) – alias name or dynamic import syntax module:class

• backend (str) – backend name e.g., chainer or pytorch

Returns:

OptimizerFactoryInterface or FunctionalOptimizerAdaptor

espnet.optimizer.pytorch

PyTorch optimizer builders.

class espnet.optimizer.pytorch.AdadeltaFactory

Bases: espnet.optimizer.factory.OptimizerFactoryInterface

Adadelta factory.

static add_arguments(parser: argparse.ArgumentParser) → argparse.ArgumentParser

Register args.

static from_args(target, args: argparse.Namespace)

Initialize optimizer from argparse Namespace.

Parameters:

• target – for pytorch model.parameters(), for chainer model

• args (argparse.Namespace) – parsed command-line args

class espnet.optimizer.pytorch.AdamFactory

Bases: espnet.optimizer.factory.OptimizerFactoryInterface

Adam factory.

static add_arguments(parser: argparse.ArgumentParser) → argparse.ArgumentParser

Register args.

static from_args(target, args: argparse.Namespace)

Initialize optimizer from argparse Namespace.

Parameters:

• target – for pytorch model.parameters(), for chainer model

• args (argparse.Namespace) – parsed command-line args

class espnet.optimizer.pytorch.SGDFactory

Bases: espnet.optimizer.factory.OptimizerFactoryInterface

SGD factory.

static add_arguments(parser: argparse.ArgumentParser) → argparse.ArgumentParser

Register args.

static from_args(target, args: argparse.Namespace)

Initialize optimizer from argparse Namespace.

Parameters:

• target – for pytorch model.parameters(), for chainer model

• args (argparse.Namespace) – parsed command-line args

espnet.optimizer.chainer

Chainer optimizer builders.

class espnet.optimizer.chainer.AdadeltaFactory

Bases: espnet.optimizer.factory.OptimizerFactoryInterface

Adadelta factory.

static add_arguments(parser: argparse.ArgumentParser) → argparse.ArgumentParser

Register args.

static from_args(target, args: argparse.Namespace)

Initialize optimizer from argparse Namespace.

Parameters:

• target – for pytorch model.parameters(), for chainer model

• args (argparse.Namespace) – parsed command-line args

class espnet.optimizer.chainer.AdamFactory

Bases: espnet.optimizer.factory.OptimizerFactoryInterface

Adam factory.

static add_arguments(parser: argparse.ArgumentParser) → argparse.ArgumentParser

Register args.

static from_args(target, args: argparse.Namespace)

Initialize optimizer from argparse Namespace.

Parameters:

• target – for pytorch model.parameters(), for chainer model

• args (argparse.Namespace) – parsed command-line args

class espnet.optimizer.chainer.SGDFactory

Bases: espnet.optimizer.factory.OptimizerFactoryInterface

SGD factory.

static add_arguments(parser: argparse.ArgumentParser) → argparse.ArgumentParser

Register args.

static from_args(target, args: argparse.Namespace)

Initialize optimizer from argparse Namespace.

Parameters:

• target – for pytorch model.parameters(), for chainer model

• args (argparse.Namespace) – parsed command-line args

espnet.optimizer.__init__

Initialize sub package.

espnet.optimizer.parser

Common optimizer default config for multiple backends.

espnet.optimizer.parser.adadelta(parser)

Add arguments.

espnet.optimizer.parser.adam(parser)

Add arguments.

espnet.optimizer.parser.sgd(parser)

Add arguments.