Added option to variable use_abs_loss on Linear layer of MLP