satellit_sam.pytorch

Auto-generated from satellit_sam/src/satellit_sam/pytorch.py by satellit_sam/scripts/generate_api_docs.py.

PyTorch initialization helpers for the SAM runtime.

Back to API index

Functions

init

def init() -> PytorchInstance:
Initialize PyTorch defaults and return backend availability details.

Returns

  • PyTorch runtime information including selected device and accelerator support.

Classes

PytorchInstance

Runtime details about the selected PyTorch backend device.

Attributes

  • cuda_available: bool
  • device: str
  • mps_available: bool

Methods

debug_info
def debug_info(self) -> None:

Log debug information about the PyTorch configuration.

Previoussatellit_sam.prompts Nextsatellit_sam.sam3