package torch

  1. Overview
  2. Docs
val resnet18 : ?num_classes:int -> Torch.Var_store.t -> Torch.Layer.t_with_training
val resnet34 : ?num_classes:int -> Torch.Var_store.t -> Torch.Layer.t_with_training
val resnet50 : ?num_classes:int -> Torch.Var_store.t -> Torch.Layer.t_with_training
val resnet101 : ?num_classes:int -> Torch.Var_store.t -> Torch.Layer.t_with_training
val resnet152 : ?num_classes:int -> Torch.Var_store.t -> Torch.Layer.t_with_training
OCaml

Innovation. Community. Security.