StochasticGradient is a high-level class for training neural networks, using a stochastic gradient algorithm. This class is serializable.

Subsections

  1. StochasticGradient(module, criterion)
  2. train(dataset)
  3. Parameters