Solved by tux in post #2 
              
              
                You can write your own update function for sure ! 
To update your weights, you might use the optimiser library. But you can also do it yourself. For example, you can basically code the gradient descent, the SGD or Adam using the following code. 
net = NN()
learning_rate = 0.01
for param in net.param…