Skip to content

base

Lens 

Lens(unembed, bias, n_head, d_model, d_vocab)

Bases: Module

It is important to remember that subclasses of the Lens cannot take separate arguments as they will be ignored at runtime for instantiation in the TransformerAttnInterface.

Parameters:

  • unembed –
  • bias –
  • n_head –
  • d_model –
  • d_vocab –

get_lens classmethod 

get_lens(name: str) -> type[Lens]

This takes the name of the lens and queries the registry to grab the corresponding Lens subclas.

Parameters:

  • name (str) –

    The name of the child Lens implementation.

Returns:

  • type[Lens] –

    Subclass Lens implementation.