Students,
Question 2.5. of Homework 3 has been altered to improve clarity in both the question itself and its results.
The updated question is as follows:
5. Define a fully connected network with one hidden layer containing 64 nodes, and include any other required layers. Specify the appropriate loss function and initialize an Adam optimizer with a learning rate of 0.0005. Set up a model saving checkpoint and early stopping to stop if there isn't any improvement in validation accuracy after 6 epochs. Prepare the function for printing and visualizing the metrics per epoch. (6pt)
Please either download the notebook with the updated question, or directly update question 2.5 on the previous notebook.