Hi, I post my program here https://github.com/Adoni/naive_network_embedding 65. I find the speed will slow down when I use backward() and step() function. I have been debugging for several days but I cannot find bugs anymore.