Choose between type of error to use for the error_mask

Currently the error mask uses the MSE to create the quantile mask. This might not be the most effective error to choose. 

Add options to choose between: 
`mse` 
`distance_loss` 
`total_loss` 