multigeom.Encoder

class multigeom.Encoder(*args, **kwargs)

What the pipeline asks of an encoder f: image -> responses (see the rules above). Any torch.nn.Module that maps an image batch [B, 1, 64, 64] in [-1, 1] to responses [B, N] satisfies it; n_neurons is optional (the width of the output is used otherwise). validate_encoder checks the behaviour that a Protocol cannot express.

__init__(*args, **kwargs)

Methods

__init__(*args, **kwargs)