Hi, I’m glad you found it useful and thanks for the feedback. I have updated the post to include the Jupyter notebook on github. Here’s the link: https://github.com/J-Yash/Creating-a-simple-NN-in-pytorch