Function rust_bert::resources::load_weights

source ·

pub fn load_weights(
    rp: &(impl ResourceProvider + ?Sized),
    vs: &mut VarStore,
    kind: Option<Kind>,
    device: Device
) -> Result<(), RustBertError>

Expand description

Load the provided VarStore with model weights from the provided ResourceProvider