Open drpngx opened 1 year ago
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.
if summary_utils.write_summary_every_n_steps(...
lingvo/jax/train.py
unreplicate_metrics=False
Not sure why, but
if summary_utils.write_summary_every_n_steps(...inlingvo/jax/train.pyfails because the metric is not replicated. I have setunreplicate_metrics=Falseand it works for me.