package torch

You can search for identifiers within the package.

in-package search v0.2.0

Torch bindings for OCaml

Install

sha256=ccd9ef3b630bdc7c41e363e71d8ecb86c316460cbf79afe67546c6ff22c19da4

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