Adding in a second neural network that guesses the identity of hidden pieces was key.