I have zero gradient error when I trained the model, don\'t know why I have this issue. Here is the error message that I got:
> -------------------------------