Define training and validation metrics

[dcfidalgo]

At the moment we have only one set of metrics defined in the head. This is problematic if we want to calculate the validation metrics during a training epoch (for example to allow early stopping within an epoch).

Idea: Provide some method that automatically duplicates the metrics, and call the corresponding metric in the training/validation_step.