I added a use_abs_loss variable to the mlp.linear layer so one can use the abs loss instead of the squared loss in the cost matrix