package torch

You can search for identifiers within the package.

in-package search v0.2.0

PyTorch bindings for OCaml

Install

Dune Dependency

github.com Readme Edit opam file Versions (18)

Authors

L

Laurent Mazare

Maintainers

L

Laurent Mazare <lmazare@gmail.com>

Sources

0.12.tar.gz

md5=ed7f1c706ae1d8f70c0db4c6cde1d391

sha512=cc09079eb8b1c876f7416b6446c9f5439b7fb54e82d8cbad778ef8bd59d62dc2681e84f54fc69c974a0ed67d3a435e6108b4303cc1c424b8dc89fdfa5992d9d7

doc/torch.vision/Torch_vision/Vgg/index.html

Module `Torch_vision.Vgg`Source

Source

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

Source

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

Source

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

Source

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

Source

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

Source

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

Source

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

Source

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

Source

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

package torch

Install

Dune Dependency

Authors

Maintainers

Sources

doc/torch.vision/Torch_vision/Vgg/index.html

Module Torch_vision.VggSource

Module `Torch_vision.Vgg`Source