pytorch-image-models
13 строк · 335.0 Байт
1try:2from torch import _assert3except ImportError:4def _assert(condition: bool, message: str):5assert condition, message6
7
8def _float_to_int(x: float) -> int:9"""10Symbolic tracing helper to substitute for inbuilt `int`.
11Hint: Inbuilt `int` can't accept an argument of type `Proxy`
12"""
13return int(x)14