`.item()` converted single-element tensors to Python floats, which
crashed `TensorAverageMeter.add()` when `--num-envs 1`. Use `.flatten()`
to preserve the tensor type.

Solving issue https://github.com/NVlabs/ProtoMotions/issues/185

Signed-off-by: protomotions <ctessler@nvidia.com>