tot_loss
tot_loss