Axon.predict throwing error: expected a %Nx.Tensor{} or a number, got: nil

[josevalim]

Can you provide your model? I think one of the layers is missing a configuration parameter.