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

Module `Torch_vision.Squeezenet`Source

val squeezenet1_0 : 
  Torch.Var_store.t ->
  num_classes:int ->
  Torch.Layer.t_with_training

val squeezenet1_1 : 
  Torch.Var_store.t ->
  num_classes:int ->
  Torch.Layer.t_with_training