Function format_params

utils/misc.py:5–9

Format parameter count in terms of K (1000)

(num: int)

from typing import Dict, Tuple

def format_params(num: int) -> str:
    """Format parameter count in terms of K (1000)"""
    if num >= 1000:
        return f"{num/1000:.2f}K"
    return str(num)

def count_parameters(model: nn.Module, verbose: bool = False) -> Dict[str, float]:
    """