Save only on main if cuda is available

Extra keyword arguments supported in addition to those in `torch.nn.Conv2d`: Args: norm (nn.Module, optional): a normali

Return an activation function given a string

Load SAM 3D Body model from config paths. Uses module-level caching to avoid reloading on every call. This runs inside the isolated

A `to_tuple` function generator. It returns a function, this function will repeat the input to a tuple of length ``n`` if the input is not an

* Get 2D angle of an OpenPose segment in pixel coordinates. * Y is flipped (pixel Y↓ → MH Y↑).

Resize image tensor [B, H, W, C] to new dimensions. If method is 'lanczos', uses PIL for high-quality downsampling (prevents compression arti

This is an operation that takes a tensor from the default process group, gathers it, stacks it, then scatters it within a smaller process sub