Thanks for the article. It happened to me a few times and ended up changing the initialization. I have one question. How do you use get_weight_grad. What is the format of data, and labels?