Recommended reading: Dive into Deep Learning — Chapter 7.
Homework 4 focuses on implementing and training CNNs for image classification tasks. Complete the exercises in the provided notebook.