package torch

  1. Overview
  2. Docs
val vgg11 : Torch.Var_store.t -> num_classes:Base.int -> Torch.Layer.t_with_training
val vgg11_bn : Torch.Var_store.t -> num_classes:Base.int -> Torch.Layer.t_with_training
val vgg13 : Torch.Var_store.t -> num_classes:Base.int -> Torch.Layer.t_with_training
val vgg13_bn : Torch.Var_store.t -> num_classes:Base.int -> Torch.Layer.t_with_training
val vgg16 : Torch.Var_store.t -> num_classes:Base.int -> Torch.Layer.t_with_training
val vgg16_bn : Torch.Var_store.t -> num_classes:Base.int -> Torch.Layer.t_with_training
val vgg19 : Torch.Var_store.t -> num_classes:Base.int -> Torch.Layer.t_with_training
val vgg19_bn : Torch.Var_store.t -> num_classes:Base.int -> Torch.Layer.t_with_training
val vgg16_layers : ?max_layer:Base.int -> Torch.Var_store.t -> batch_norm:Base.bool -> (Torch.Tensor.t -> (Base.int, Torch.Tensor.t, Base.Int.comparator_witness) Base.Map.t) Base.Staged.t
OCaml

Innovation. Community. Security.