package torch

  1. Overview
  2. Docs
val densenet121 : Torch.Var_store.t -> num_classes:int -> Torch.Layer.t_with_training
val densenet161 : Torch.Var_store.t -> num_classes:int -> Torch.Layer.t_with_training
val densenet169 : Torch.Var_store.t -> num_classes:int -> Torch.Layer.t_with_training
val densenet201 : Torch.Var_store.t -> num_classes:int -> Torch.Layer.t_with_training