search

gvtulder / elasticdeform

Differentiable elastic deformations for N-dimensional images (Python, SciPy, NumPy, TensorFlow, PyTorch).
Other
186 stars 25 forks source link

No gradient #19

Open romane665 opened 6 months ago

romane665 commented 6 months ago

Hello,

I've been trying to use the function deform_grid of elasticdeform.torch for deep learning, but the back propagation doesn't seem to be working properly. When i model.print_gradients(), i get "No gradient". Has anyone encountered a similar issue ?

Thanks in advance.