[][src]Function tch::vision::resnet::resnet18

pub fn resnet18(p: &Path, num_classes: i64) -> FuncT<'static>

Creates a ResNet-18 model.

Pre-trained weights can be downloaded at the following link: https://github.com/LaurentMazare/ocaml-torch/releases/download/v0.1-unstable/resnet18.ot