MLPerf block_stop metadata fix: s/epoch_num/epoch_count
MLPerf block_stop metadata fix: s/epoch_num/epoch_count

PiperOrigin-RevId: 401616375

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo