Function fix

opacus/validators/lstm.py:38–49

(module: nn.LSTM)

@register_module_fixer(nn.LSTM)
def fix(module: nn.LSTM) -> DPLSTM:
    dplstm = DPLSTM(
        input_size=module.input_size,
        hidden_size=module.hidden_size,
        num_layers=module.num_layers,
        bias=module.bias,
        batch_first=module.batch_first,
        dropout=module.dropout,
        bidirectional=module.bidirectional,
    )
    dplstm.load_state_dict(module.state_dict())
    return dplstm