Allow for more than one output for loss and metric