Revert "Stop reseting the neural network"

This reverts commit 0cc8d849da8c05338e28d7e4987c9e192017e7ab.