Gradient mismatch is serial and parallel

There are way to many scatter reverses in your code.
See for instance

The current code you have shared has so many components that I don’t have time to go through them carefully in detail.

1 Like