The setup function is used to set important attributes and method for
to be used with Luz.
setup(module, loss = NULL, optimizer = NULL, metrics = NULL)
nn_module that you want set up.
function, optional) An optional function with the signature
function(input, target). It's only requires if your
implement a method called
torch_optimizer, optional) A function with the signature
function(parameters, ...) that is used to initialize an optimizer given
the model parameters.
list, optional) A list of metrics to be tracked during
the training procedure.
A luz module that can be trained with
It makes sure the module have all the necessary ingredients in order to be fitted.