Hi @JuanFMontesinos,
Thanks for your reply! When you say to check if gradients are flowing back from input, do you mean I should do something like
for p in model.parameters():
    print(p.grad)

Will this tell me if gradients are flowing backwards in a proper way?
If things are going wrong should I get something like zero gradients?