write_checkpoint

Function write_checkpoint 

Source
pub fn write_checkpoint<B: Backend>(
    writer: &mut impl Write,
    tensors: &[(String, Tensor<B>)],
) -> Result<()>
Expand description

Write a set of named tensors to a writer in the Shrew checkpoint format.