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.
@callback architectures() :: [atom()]
Returns the list of supported model architectures.
Builds a template input for the model.
The template is used to compile the model when initializing parameters.
Axon model according to the given configuration.