read_torch_pt

Load PyTorch .pt file (for conds.pt)

Description

This is a simplified loader for the specific structure of conds.pt Full pickle parsing is complex; this handles the common case

Usage

1read_torch_pt(path, device = "cpu")

Arguments

  • path: Path to .pt file
  • device: Device to load to

Value

Loaded object (typically a list/dict structure)