`olm.nn.torch_nn_wrappers`

Source: src/olm/nn/torch_nn_wrappers.py:1

Thin wrappers around torch.nn modules.

Example

Block([
    Embedding(vocab_size, embed_dim),
    AbsolutePositionalEmbedding(max_seq_len, embed_dim, dropout),
])

Classes

Linear(*args, **kwargs)

Bases: Linear

Source: src/olm/nn/torch_nn_wrappers.py:12

Methods

forward(self, x)

Source: src/olm/nn/torch_nn_wrappers.py:16