I have been exercising in building a simple NN from scratch and I am having the following recurrent issue with the code below.