If you run the chapter 3 code notebook, the validation loss stops after one epoch after the loss goes up.