Up – Package index » torch » Library torch.vision » Torch_vision » Densenet

Module `Torch_vision.Densenet`Source

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