ColossalAI/colossalai/legacy/trainer/hooks/_commons_.py

import torch


def _format_number(val, prec=5):
    if isinstance(val, float):
        return f"{val:.{prec}g}"
    elif torch.is_tensor(val) and torch.is_floating_point(val):
        return f"{val.item():.{prec}g}"
    return val