olm.nn.torch_nn_wrappers¶
Thin wrappers around torch.nn modules.
Example::
: Block([
: Embedding(vocab_size, embed_dim),
AbsolutePositionalEmbedding(max_seq_len, embed_dim, dropout),
])
Classes¶
Linear(*args, **kwargs) |
|
|---|---|
class olm.nn.torch_nn_wrappers.Linear(*args: Any, **kwargs: Any)¶
Bases: Linear