You are likely running into this issue causing the second backward call to fail as the parameters were updated resulting in stale forward activations.
You are likely running into this issue causing the second backward call to fail as the parameters were updated resulting in stale forward activations.