Source file torch.ml

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
module Checkpointing = Checkpointing
module Cifar_helper = Cifar_helper
module Cuda = Cuda
module Dataset_helper = Dataset_helper
module Device = Device
module Ivalue = Ivalue
module Layer = Layer
module Mnist_helper = Mnist_helper
module Module = Module

(* [Nn] is an alias to [Layer] to keep coherence with the pytorch names. *)
module Nn = Layer
module Optimizer = Optimizer
module Scalar = Scalar
module Serialize = Serialize
module Tensor = Tensor
module Text_helper = Text_helper
module Var_store = Var_store