Cell no. 3, showing error

Co-authored-by: Hamid Shojanazeri <hamid.nazeri2010@gmail.com>

Co-authored-by: Hamid Shojanazeri <hamid.nazeri2010@gmail.com>

Co-authored-by: Hamid Shojanazeri <hamid.nazeri2010@gmail.com>

**Problem**
Currently, we're val_loss.append(best_val_loss) in each epoch. This is misleading because we're appending the corresponding epoch (not best across epochs) quantities in train_loss, train_prep, and val_prep. This is also inconvenient, as one often would like to plot both train and validation losses as a function of the epochs to look for overfitting.

**Solution**
val_loss.append(eval_epoch_loss)