Store FLOS as floats to avoid overflow.
# What does this PR do?

As pointed out in #10212, the `total_flos` stored as ints can result in overflowing errors: when in Python ints there is no risk, but when in distributed training, we use torch.int64 to gather all FLOS on all processes which can trigger that error.

Fixes #10212