Function candle_core::pickle::read_pth_tensor_info

source · 
pub fn read_pth_tensor_info<P: AsRef<Path>>(
    file: P,
    verbose: bool,
    key: Option<&str>
) -> Result<Vec<TensorInfo>>
Expand description

Read the tensor info from a .pth file.

§Arguments

  • file - The path to the .pth file.
  • verbose - Whether to print debug information.
  • key - Optional key to retrieve state_dict from the pth file.