Fix a bug in eval summary writing in the classification trainer. `log_eval_summary` should be called after the loop over eval batches.
Fix a bug in eval summary writing in the classification trainer. `log_eval_summary` should be called after the loop over eval batches.

PiperOrigin-RevId: 417783375

google-research/scenic