Validation Loss doesn't go down in Chapter 3 Code

[davidfufu]

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