“The accuracy of predicting environment parameters, such as cube size, from a policy's memory state showed a positive correlation with the diversity of the training environment.”