aamini / introtodeeplearning

Lab Materials for MIT 6.S191: Introduction to Deep Learning
MIT License
7.19k stars 3.63k forks source link

Unpack return value in Lab 2 Part 2 #141

Open yufengzh opened 1 year ago

yufengzh commented 1 year ago

The ss_vae_loss_function returns a length-3 tuple, namely loss, class_loss, and vae_loss. Therefore, we add a third one when calling this function.