Given a state, return the data which is input into the model.
BaseAlphaModule.get_data()
dgs.models.alpha.alpha.BaseAlphaModule.forward
dgs.models.alpha.alpha.BaseAlphaModule.load_weights