search

NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
https://nvidia.github.io/MinkowskiEngine
Other
2.47k stars 365 forks source link

A question about how to call the backward propagated hook function #399

Open QWTforGithub opened 3 years ago

QWTforGithub commented 3 years ago

I created a network with MinkowskiEngin and bound a backpropagation hook function to one layer of the network. When I get the result of the network, I explicitly propagate the result backward, but I cannot call the backward propagated hook function of that layer. How can I call this hook function when propagating backward?

QWTforGithub commented 3 years ago

I created a network with MinkowskiEngin and bound a backpropagation hook function to one layer of the network. When I get the result of the network, I explicitly propagate the result backward, but I cannot call the backward propagated hook function of that layer. How can I call this hook function when propagating backward?

My goal was to get the gradient of that layer, but I found that the hook function of that layer was not called when backward propagation occurred.