espnet2.asr.state_spaces.s4.OptimModule

source

class espnet2.asr.state_spaces.s4.OptimModule(*args, **kwargs)

Bases: Module

Interface for Module that allows registering buffers/parameters with configurable optimizer hyperparameters. # noqa

Initializes internal Module state, shared by both nn.Module and ScriptModule.

register(name, tensor, lr=None)

Register a tensor with a configurable learning rate and 0 weight decay.