read_checkpoint

Function read_checkpoint 

Source
pub fn read_checkpoint<B: Backend>(
    reader: &mut impl Read,
    device: &B::Device,
) -> Result<Vec<(String, Tensor<B>)>>
Expand description

Read named tensors from a reader in the Shrew checkpoint format.