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