In: https://github.com/MITDeepLearning/introtodeeplearning/blob/master/lab2/solutions/PT_Part2_Debiasing_Solution.ipynb According to the description, the calculation should be: total_loss = torch.mean(classification_loss + face_indicator * vae_loss) Correct me if I'm wrong, Thanks! 