unreplicate_metrics=True fails on my training #331
Description
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.