Test
total_loss