I found some bugs in the backprop implementation