pub fn load_linear<B: Backend>(
    path: &str,
    device: &B::Device
) -> Result<Linear<B>, Box<dyn Error>>