added function to model config to query training and test stats.
added function to model config to query training and test stats.

PiperOrigin-RevId: 251266700

google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research