unreplicate_metrics=True fails on my training

[drpngx]

Not sure why, but if summary_utils.write_summary_every_n_steps(... in lingvo/jax/train.py fails because the metric is not replicated. I have set unreplicate_metrics=False and it works for me.