Custom Layer Grad is NoneType

To anyone looking in the future. See this answer and this page.

1 Like