View Source Bumblebee.ModelSpec behaviour (Bumblebee v0.4.2)

An interface for configuring and building models based on the same architecture.

Every module implementing this behaviour is expected to also define a configuration struct.

Summary

t()

architectures()

Returns the list of supported model architectures.

Builds a template input for the model.

model(t)

Builds an Axon model according to the given configuration.

@type t() :: Bumblebee.Configurable.t()

@callback architectures() :: [atom()]

Returns the list of supported model architectures.

@callback input_template(t()) :: map()

Builds a template input for the model.

The template is used to compile the model when initializing parameters.

@callback model(t()) :: Axon.t()

Builds an Axon model according to the given configuration.