Function convert_unit

tokasaurus/utils.py:188–196

(milis: float, unit: str)

def convert_unit(milis: float, unit: str) -> float:
    if unit == "ms":
        return milis
    elif unit == "s":
        return milis / 1000
    elif unit == "us":
        return milis * 1000
    else:
        raise ValueError(f"Invalid unit: {unit}")


@torch.no_grad()